/* Crypto-pattern optimizer.
   Copyright (C) 2003-2023 Free Software Foundation, Inc.

This file is part of GCC.

GCC is free software; you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by the Free
Software Foundation; either version 3, or (at your option) any later
version.

GCC is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
for more details.

You should have received a copy of the GNU General Public License
along with GCC; see the file COPYING3.  If not see
<http://www.gnu.org/licenses/>.  */

#include "config.h"
#define INCLUDE_VECTOR
#define INCLUDE_MAP
#define INCLUDE_SET
#define INCLUDE_ALGORITHM
#include "system.h"
#include "coretypes.h"
#include "backend.h"
#include "target.h"
#include "rtl.h"
#include "tree.h"
#include "df.h"
#include "memmodel.h"
#include "optabs.h"
#include "regs.h"
#include "emit-rtl.h"
#include "recog.h"
#include "cfgrtl.h"
#include "cfgcleanup.h"
#include "expr.h"
#include "tree-pass.h"
#include "rtl-matcher.h"

/* Basic AES table descryption.  */
struct aes_table
{
  /* Number of elements per table.  */
  static const unsigned int table_nelts = 256;
  /* Number of tables.  */
  static const unsigned int basic_tables_num = 4;
  /* Number of rounds.  */
  static const unsigned int rounds_num = 4;
  /* Common ID for wrong table.  */
  static const unsigned int BAD_TABLE = -1;

  typedef const unsigned int table_type[table_nelts];
  typedef table_type *table_map[basic_tables_num];

  template<typename T>
  static bool is_basic_table (tree ctor, const T ethalon[table_nelts])
    {
      if (TREE_CODE (ctor) != CONSTRUCTOR
	  ||CONSTRUCTOR_NELTS (ctor) != table_nelts)
	return false;

      unsigned ix;
      tree val;
      FOR_EACH_CONSTRUCTOR_VALUE (CONSTRUCTOR_ELTS (ctor), ix, val)
	if (TREE_INT_CST_LOW (val) != ethalon[ix])
	  return false;
      return true;
    }

  static unsigned check_table (tree ctor,
			       table_map tables)
    {
      for (unsigned i = 0; i < 4; ++i)
	if (is_basic_table (ctor, *tables[i]))
	  return i;
      return BAD_TABLE;
    }
};

/* AES encryption info.  */
struct aes_encrypt_table : aes_table
{
  typedef enum
  {
    TE0,
    TE1,
    TE2,
    TE3,
    BAD_TABLE = aes_table::BAD_TABLE
  } table_entry;

  static table_type Te0;
  static table_type Te1;
  static table_type Te2;
  static table_type Te3;

  static table_map tables;
  static table_entry rounds[rounds_num];
  static table_entry final_rounds[rounds_num];

  static table_entry get_table_id (tree ctor)
    {
      return static_cast<table_entry> (check_table (ctor, tables));
    }
};

/* AES decryption info.  */
struct aes_decrypt_table : aes_table
{
  typedef enum
  {
    TD0,
    TD1,
    TD2,
    TD3,
    TD4,
    BAD_TABLE = aes_table::BAD_TABLE
  } table_entry;

  static table_type Td0;
  static table_type Td1;
  static table_type Td2;
  static table_type Td3;

  static table_map tables;
  static table_entry rounds[rounds_num];
  static table_entry final_rounds[rounds_num];

  static const unsigned char Td4[table_nelts];

  /* TD4 requires special handler due to type shrinking optimizations.  */
  static bool is_td4 (tree ctor)
    {
      if (is_basic_table (ctor, Td4))
	return true;

      if (TREE_CODE (ctor) != STRING_CST
	  || TREE_STRING_LENGTH (ctor) != table_nelts)
	return false;

      const unsigned char *p
	= (const unsigned char *) TREE_STRING_POINTER (ctor);
      for (int i = 0; i < TREE_STRING_LENGTH (ctor); ++i)
	if (p[i] != Td4[i])
	  return false;

      return true;
    }

  static table_entry get_table_id (tree ctor)
    {
      unsigned int res = check_table (ctor, tables);
      if (res == aes_table::BAD_TABLE
	  && is_td4 (ctor))
	return TD4;
      return static_cast<table_entry> (res);
    }
};

/* Basic tables info.  */
aes_encrypt_table::table_map aes_encrypt_table::tables
  = { &Te0, &Te1, &Te2, &Te3 };
aes_decrypt_table::table_map aes_decrypt_table::tables
  = { &Td0, &Td1, &Td2, &Td3 };

/* Round tables permutations info.  */
aes_encrypt_table::table_entry aes_encrypt_table::rounds[]
  = {TE0, TE1, TE2, TE3};
aes_decrypt_table::table_entry aes_decrypt_table::rounds[]
  = {TD0, TD1, TD2, TD3};
aes_encrypt_table::table_entry aes_encrypt_table::final_rounds[]
  = {TE2, TE3, TE0, TE1};
aes_decrypt_table::table_entry aes_decrypt_table::final_rounds[]
  = {TD4, TD4, TD4, TD4};

aes_encrypt_table::table_type aes_encrypt_table::Te0 = {
    0xc66363a5U, 0xf87c7c84U, 0xee777799U, 0xf67b7b8dU,
    0xfff2f20dU, 0xd66b6bbdU, 0xde6f6fb1U, 0x91c5c554U,
    0x60303050U, 0x02010103U, 0xce6767a9U, 0x562b2b7dU,
    0xe7fefe19U, 0xb5d7d762U, 0x4dababe6U, 0xec76769aU,
    0x8fcaca45U, 0x1f82829dU, 0x89c9c940U, 0xfa7d7d87U,
    0xeffafa15U, 0xb25959ebU, 0x8e4747c9U, 0xfbf0f00bU,
    0x41adadecU, 0xb3d4d467U, 0x5fa2a2fdU, 0x45afafeaU,
    0x239c9cbfU, 0x53a4a4f7U, 0xe4727296U, 0x9bc0c05bU,
    0x75b7b7c2U, 0xe1fdfd1cU, 0x3d9393aeU, 0x4c26266aU,
    0x6c36365aU, 0x7e3f3f41U, 0xf5f7f702U, 0x83cccc4fU,
    0x6834345cU, 0x51a5a5f4U, 0xd1e5e534U, 0xf9f1f108U,
    0xe2717193U, 0xabd8d873U, 0x62313153U, 0x2a15153fU,
    0x0804040cU, 0x95c7c752U, 0x46232365U, 0x9dc3c35eU,
    0x30181828U, 0x379696a1U, 0x0a05050fU, 0x2f9a9ab5U,
    0x0e070709U, 0x24121236U, 0x1b80809bU, 0xdfe2e23dU,
    0xcdebeb26U, 0x4e272769U, 0x7fb2b2cdU, 0xea75759fU,
    0x1209091bU, 0x1d83839eU, 0x582c2c74U, 0x341a1a2eU,
    0x361b1b2dU, 0xdc6e6eb2U, 0xb45a5aeeU, 0x5ba0a0fbU,
    0xa45252f6U, 0x763b3b4dU, 0xb7d6d661U, 0x7db3b3ceU,
    0x5229297bU, 0xdde3e33eU, 0x5e2f2f71U, 0x13848497U,
    0xa65353f5U, 0xb9d1d168U, 0x00000000U, 0xc1eded2cU,
    0x40202060U, 0xe3fcfc1fU, 0x79b1b1c8U, 0xb65b5bedU,
    0xd46a6abeU, 0x8dcbcb46U, 0x67bebed9U, 0x7239394bU,
    0x944a4adeU, 0x984c4cd4U, 0xb05858e8U, 0x85cfcf4aU,
    0xbbd0d06bU, 0xc5efef2aU, 0x4faaaae5U, 0xedfbfb16U,
    0x864343c5U, 0x9a4d4dd7U, 0x66333355U, 0x11858594U,
    0x8a4545cfU, 0xe9f9f910U, 0x04020206U, 0xfe7f7f81U,
    0xa05050f0U, 0x783c3c44U, 0x259f9fbaU, 0x4ba8a8e3U,
    0xa25151f3U, 0x5da3a3feU, 0x804040c0U, 0x058f8f8aU,
    0x3f9292adU, 0x219d9dbcU, 0x70383848U, 0xf1f5f504U,
    0x63bcbcdfU, 0x77b6b6c1U, 0xafdada75U, 0x42212163U,
    0x20101030U, 0xe5ffff1aU, 0xfdf3f30eU, 0xbfd2d26dU,
    0x81cdcd4cU, 0x180c0c14U, 0x26131335U, 0xc3ecec2fU,
    0xbe5f5fe1U, 0x359797a2U, 0x884444ccU, 0x2e171739U,
    0x93c4c457U, 0x55a7a7f2U, 0xfc7e7e82U, 0x7a3d3d47U,
    0xc86464acU, 0xba5d5de7U, 0x3219192bU, 0xe6737395U,
    0xc06060a0U, 0x19818198U, 0x9e4f4fd1U, 0xa3dcdc7fU,
    0x44222266U, 0x542a2a7eU, 0x3b9090abU, 0x0b888883U,
    0x8c4646caU, 0xc7eeee29U, 0x6bb8b8d3U, 0x2814143cU,
    0xa7dede79U, 0xbc5e5ee2U, 0x160b0b1dU, 0xaddbdb76U,
    0xdbe0e03bU, 0x64323256U, 0x743a3a4eU, 0x140a0a1eU,
    0x924949dbU, 0x0c06060aU, 0x4824246cU, 0xb85c5ce4U,
    0x9fc2c25dU, 0xbdd3d36eU, 0x43acacefU, 0xc46262a6U,
    0x399191a8U, 0x319595a4U, 0xd3e4e437U, 0xf279798bU,
    0xd5e7e732U, 0x8bc8c843U, 0x6e373759U, 0xda6d6db7U,
    0x018d8d8cU, 0xb1d5d564U, 0x9c4e4ed2U, 0x49a9a9e0U,
    0xd86c6cb4U, 0xac5656faU, 0xf3f4f407U, 0xcfeaea25U,
    0xca6565afU, 0xf47a7a8eU, 0x47aeaee9U, 0x10080818U,
    0x6fbabad5U, 0xf0787888U, 0x4a25256fU, 0x5c2e2e72U,
    0x381c1c24U, 0x57a6a6f1U, 0x73b4b4c7U, 0x97c6c651U,
    0xcbe8e823U, 0xa1dddd7cU, 0xe874749cU, 0x3e1f1f21U,
    0x964b4bddU, 0x61bdbddcU, 0x0d8b8b86U, 0x0f8a8a85U,
    0xe0707090U, 0x7c3e3e42U, 0x71b5b5c4U, 0xcc6666aaU,
    0x904848d8U, 0x06030305U, 0xf7f6f601U, 0x1c0e0e12U,
    0xc26161a3U, 0x6a35355fU, 0xae5757f9U, 0x69b9b9d0U,
    0x17868691U, 0x99c1c158U, 0x3a1d1d27U, 0x279e9eb9U,
    0xd9e1e138U, 0xebf8f813U, 0x2b9898b3U, 0x22111133U,
    0xd26969bbU, 0xa9d9d970U, 0x078e8e89U, 0x339494a7U,
    0x2d9b9bb6U, 0x3c1e1e22U, 0x15878792U, 0xc9e9e920U,
    0x87cece49U, 0xaa5555ffU, 0x50282878U, 0xa5dfdf7aU,
    0x038c8c8fU, 0x59a1a1f8U, 0x09898980U, 0x1a0d0d17U,
    0x65bfbfdaU, 0xd7e6e631U, 0x844242c6U, 0xd06868b8U,
    0x824141c3U, 0x299999b0U, 0x5a2d2d77U, 0x1e0f0f11U,
    0x7bb0b0cbU, 0xa85454fcU, 0x6dbbbbd6U, 0x2c16163aU,
};

aes_encrypt_table::table_type aes_encrypt_table::Te1 = {
    0xa5c66363U, 0x84f87c7cU, 0x99ee7777U, 0x8df67b7bU,
    0x0dfff2f2U, 0xbdd66b6bU, 0xb1de6f6fU, 0x5491c5c5U,
    0x50603030U, 0x03020101U, 0xa9ce6767U, 0x7d562b2bU,
    0x19e7fefeU, 0x62b5d7d7U, 0xe64dababU, 0x9aec7676U,
    0x458fcacaU, 0x9d1f8282U, 0x4089c9c9U, 0x87fa7d7dU,
    0x15effafaU, 0xebb25959U, 0xc98e4747U, 0x0bfbf0f0U,
    0xec41adadU, 0x67b3d4d4U, 0xfd5fa2a2U, 0xea45afafU,
    0xbf239c9cU, 0xf753a4a4U, 0x96e47272U, 0x5b9bc0c0U,
    0xc275b7b7U, 0x1ce1fdfdU, 0xae3d9393U, 0x6a4c2626U,
    0x5a6c3636U, 0x417e3f3fU, 0x02f5f7f7U, 0x4f83ccccU,
    0x5c683434U, 0xf451a5a5U, 0x34d1e5e5U, 0x08f9f1f1U,
    0x93e27171U, 0x73abd8d8U, 0x53623131U, 0x3f2a1515U,
    0x0c080404U, 0x5295c7c7U, 0x65462323U, 0x5e9dc3c3U,
    0x28301818U, 0xa1379696U, 0x0f0a0505U, 0xb52f9a9aU,
    0x090e0707U, 0x36241212U, 0x9b1b8080U, 0x3ddfe2e2U,
    0x26cdebebU, 0x694e2727U, 0xcd7fb2b2U, 0x9fea7575U,
    0x1b120909U, 0x9e1d8383U, 0x74582c2cU, 0x2e341a1aU,
    0x2d361b1bU, 0xb2dc6e6eU, 0xeeb45a5aU, 0xfb5ba0a0U,
    0xf6a45252U, 0x4d763b3bU, 0x61b7d6d6U, 0xce7db3b3U,
    0x7b522929U, 0x3edde3e3U, 0x715e2f2fU, 0x97138484U,
    0xf5a65353U, 0x68b9d1d1U, 0x00000000U, 0x2cc1ededU,
    0x60402020U, 0x1fe3fcfcU, 0xc879b1b1U, 0xedb65b5bU,
    0xbed46a6aU, 0x468dcbcbU, 0xd967bebeU, 0x4b723939U,
    0xde944a4aU, 0xd4984c4cU, 0xe8b05858U, 0x4a85cfcfU,
    0x6bbbd0d0U, 0x2ac5efefU, 0xe54faaaaU, 0x16edfbfbU,
    0xc5864343U, 0xd79a4d4dU, 0x55663333U, 0x94118585U,
    0xcf8a4545U, 0x10e9f9f9U, 0x06040202U, 0x81fe7f7fU,
    0xf0a05050U, 0x44783c3cU, 0xba259f9fU, 0xe34ba8a8U,
    0xf3a25151U, 0xfe5da3a3U, 0xc0804040U, 0x8a058f8fU,
    0xad3f9292U, 0xbc219d9dU, 0x48703838U, 0x04f1f5f5U,
    0xdf63bcbcU, 0xc177b6b6U, 0x75afdadaU, 0x63422121U,
    0x30201010U, 0x1ae5ffffU, 0x0efdf3f3U, 0x6dbfd2d2U,
    0x4c81cdcdU, 0x14180c0cU, 0x35261313U, 0x2fc3ececU,
    0xe1be5f5fU, 0xa2359797U, 0xcc884444U, 0x392e1717U,
    0x5793c4c4U, 0xf255a7a7U, 0x82fc7e7eU, 0x477a3d3dU,
    0xacc86464U, 0xe7ba5d5dU, 0x2b321919U, 0x95e67373U,
    0xa0c06060U, 0x98198181U, 0xd19e4f4fU, 0x7fa3dcdcU,
    0x66442222U, 0x7e542a2aU, 0xab3b9090U, 0x830b8888U,
    0xca8c4646U, 0x29c7eeeeU, 0xd36bb8b8U, 0x3c281414U,
    0x79a7dedeU, 0xe2bc5e5eU, 0x1d160b0bU, 0x76addbdbU,
    0x3bdbe0e0U, 0x56643232U, 0x4e743a3aU, 0x1e140a0aU,
    0xdb924949U, 0x0a0c0606U, 0x6c482424U, 0xe4b85c5cU,
    0x5d9fc2c2U, 0x6ebdd3d3U, 0xef43acacU, 0xa6c46262U,
    0xa8399191U, 0xa4319595U, 0x37d3e4e4U, 0x8bf27979U,
    0x32d5e7e7U, 0x438bc8c8U, 0x596e3737U, 0xb7da6d6dU,
    0x8c018d8dU, 0x64b1d5d5U, 0xd29c4e4eU, 0xe049a9a9U,
    0xb4d86c6cU, 0xfaac5656U, 0x07f3f4f4U, 0x25cfeaeaU,
    0xafca6565U, 0x8ef47a7aU, 0xe947aeaeU, 0x18100808U,
    0xd56fbabaU, 0x88f07878U, 0x6f4a2525U, 0x725c2e2eU,
    0x24381c1cU, 0xf157a6a6U, 0xc773b4b4U, 0x5197c6c6U,
    0x23cbe8e8U, 0x7ca1ddddU, 0x9ce87474U, 0x213e1f1fU,
    0xdd964b4bU, 0xdc61bdbdU, 0x860d8b8bU, 0x850f8a8aU,
    0x90e07070U, 0x427c3e3eU, 0xc471b5b5U, 0xaacc6666U,
    0xd8904848U, 0x05060303U, 0x01f7f6f6U, 0x121c0e0eU,
    0xa3c26161U, 0x5f6a3535U, 0xf9ae5757U, 0xd069b9b9U,
    0x91178686U, 0x5899c1c1U, 0x273a1d1dU, 0xb9279e9eU,
    0x38d9e1e1U, 0x13ebf8f8U, 0xb32b9898U, 0x33221111U,
    0xbbd26969U, 0x70a9d9d9U, 0x89078e8eU, 0xa7339494U,
    0xb62d9b9bU, 0x223c1e1eU, 0x92158787U, 0x20c9e9e9U,
    0x4987ceceU, 0xffaa5555U, 0x78502828U, 0x7aa5dfdfU,
    0x8f038c8cU, 0xf859a1a1U, 0x80098989U, 0x171a0d0dU,
    0xda65bfbfU, 0x31d7e6e6U, 0xc6844242U, 0xb8d06868U,
    0xc3824141U, 0xb0299999U, 0x775a2d2dU, 0x111e0f0fU,
    0xcb7bb0b0U, 0xfca85454U, 0xd66dbbbbU, 0x3a2c1616U,
};

aes_encrypt_table::table_type aes_encrypt_table::Te2 = {
    0x63a5c663U, 0x7c84f87cU, 0x7799ee77U, 0x7b8df67bU,
    0xf20dfff2U, 0x6bbdd66bU, 0x6fb1de6fU, 0xc55491c5U,
    0x30506030U, 0x01030201U, 0x67a9ce67U, 0x2b7d562bU,
    0xfe19e7feU, 0xd762b5d7U, 0xabe64dabU, 0x769aec76U,
    0xca458fcaU, 0x829d1f82U, 0xc94089c9U, 0x7d87fa7dU,
    0xfa15effaU, 0x59ebb259U, 0x47c98e47U, 0xf00bfbf0U,
    0xadec41adU, 0xd467b3d4U, 0xa2fd5fa2U, 0xafea45afU,
    0x9cbf239cU, 0xa4f753a4U, 0x7296e472U, 0xc05b9bc0U,
    0xb7c275b7U, 0xfd1ce1fdU, 0x93ae3d93U, 0x266a4c26U,
    0x365a6c36U, 0x3f417e3fU, 0xf702f5f7U, 0xcc4f83ccU,
    0x345c6834U, 0xa5f451a5U, 0xe534d1e5U, 0xf108f9f1U,
    0x7193e271U, 0xd873abd8U, 0x31536231U, 0x153f2a15U,
    0x040c0804U, 0xc75295c7U, 0x23654623U, 0xc35e9dc3U,
    0x18283018U, 0x96a13796U, 0x050f0a05U, 0x9ab52f9aU,
    0x07090e07U, 0x12362412U, 0x809b1b80U, 0xe23ddfe2U,
    0xeb26cdebU, 0x27694e27U, 0xb2cd7fb2U, 0x759fea75U,
    0x091b1209U, 0x839e1d83U, 0x2c74582cU, 0x1a2e341aU,
    0x1b2d361bU, 0x6eb2dc6eU, 0x5aeeb45aU, 0xa0fb5ba0U,
    0x52f6a452U, 0x3b4d763bU, 0xd661b7d6U, 0xb3ce7db3U,
    0x297b5229U, 0xe33edde3U, 0x2f715e2fU, 0x84971384U,
    0x53f5a653U, 0xd168b9d1U, 0x00000000U, 0xed2cc1edU,
    0x20604020U, 0xfc1fe3fcU, 0xb1c879b1U, 0x5bedb65bU,
    0x6abed46aU, 0xcb468dcbU, 0xbed967beU, 0x394b7239U,
    0x4ade944aU, 0x4cd4984cU, 0x58e8b058U, 0xcf4a85cfU,
    0xd06bbbd0U, 0xef2ac5efU, 0xaae54faaU, 0xfb16edfbU,
    0x43c58643U, 0x4dd79a4dU, 0x33556633U, 0x85941185U,
    0x45cf8a45U, 0xf910e9f9U, 0x02060402U, 0x7f81fe7fU,
    0x50f0a050U, 0x3c44783cU, 0x9fba259fU, 0xa8e34ba8U,
    0x51f3a251U, 0xa3fe5da3U, 0x40c08040U, 0x8f8a058fU,
    0x92ad3f92U, 0x9dbc219dU, 0x38487038U, 0xf504f1f5U,
    0xbcdf63bcU, 0xb6c177b6U, 0xda75afdaU, 0x21634221U,
    0x10302010U, 0xff1ae5ffU, 0xf30efdf3U, 0xd26dbfd2U,
    0xcd4c81cdU, 0x0c14180cU, 0x13352613U, 0xec2fc3ecU,
    0x5fe1be5fU, 0x97a23597U, 0x44cc8844U, 0x17392e17U,
    0xc45793c4U, 0xa7f255a7U, 0x7e82fc7eU, 0x3d477a3dU,
    0x64acc864U, 0x5de7ba5dU, 0x192b3219U, 0x7395e673U,
    0x60a0c060U, 0x81981981U, 0x4fd19e4fU, 0xdc7fa3dcU,
    0x22664422U, 0x2a7e542aU, 0x90ab3b90U, 0x88830b88U,
    0x46ca8c46U, 0xee29c7eeU, 0xb8d36bb8U, 0x143c2814U,
    0xde79a7deU, 0x5ee2bc5eU, 0x0b1d160bU, 0xdb76addbU,
    0xe03bdbe0U, 0x32566432U, 0x3a4e743aU, 0x0a1e140aU,
    0x49db9249U, 0x060a0c06U, 0x246c4824U, 0x5ce4b85cU,
    0xc25d9fc2U, 0xd36ebdd3U, 0xacef43acU, 0x62a6c462U,
    0x91a83991U, 0x95a43195U, 0xe437d3e4U, 0x798bf279U,
    0xe732d5e7U, 0xc8438bc8U, 0x37596e37U, 0x6db7da6dU,
    0x8d8c018dU, 0xd564b1d5U, 0x4ed29c4eU, 0xa9e049a9U,
    0x6cb4d86cU, 0x56faac56U, 0xf407f3f4U, 0xea25cfeaU,
    0x65afca65U, 0x7a8ef47aU, 0xaee947aeU, 0x08181008U,
    0xbad56fbaU, 0x7888f078U, 0x256f4a25U, 0x2e725c2eU,
    0x1c24381cU, 0xa6f157a6U, 0xb4c773b4U, 0xc65197c6U,
    0xe823cbe8U, 0xdd7ca1ddU, 0x749ce874U, 0x1f213e1fU,
    0x4bdd964bU, 0xbddc61bdU, 0x8b860d8bU, 0x8a850f8aU,
    0x7090e070U, 0x3e427c3eU, 0xb5c471b5U, 0x66aacc66U,
    0x48d89048U, 0x03050603U, 0xf601f7f6U, 0x0e121c0eU,
    0x61a3c261U, 0x355f6a35U, 0x57f9ae57U, 0xb9d069b9U,
    0x86911786U, 0xc15899c1U, 0x1d273a1dU, 0x9eb9279eU,
    0xe138d9e1U, 0xf813ebf8U, 0x98b32b98U, 0x11332211U,
    0x69bbd269U, 0xd970a9d9U, 0x8e89078eU, 0x94a73394U,
    0x9bb62d9bU, 0x1e223c1eU, 0x87921587U, 0xe920c9e9U,
    0xce4987ceU, 0x55ffaa55U, 0x28785028U, 0xdf7aa5dfU,
    0x8c8f038cU, 0xa1f859a1U, 0x89800989U, 0x0d171a0dU,
    0xbfda65bfU, 0xe631d7e6U, 0x42c68442U, 0x68b8d068U,
    0x41c38241U, 0x99b02999U, 0x2d775a2dU, 0x0f111e0fU,
    0xb0cb7bb0U, 0x54fca854U, 0xbbd66dbbU, 0x163a2c16U,
};

aes_encrypt_table::table_type aes_encrypt_table::Te3 = {
    0x6363a5c6U, 0x7c7c84f8U, 0x777799eeU, 0x7b7b8df6U,
    0xf2f20dffU, 0x6b6bbdd6U, 0x6f6fb1deU, 0xc5c55491U,
    0x30305060U, 0x01010302U, 0x6767a9ceU, 0x2b2b7d56U,
    0xfefe19e7U, 0xd7d762b5U, 0xababe64dU, 0x76769aecU,
    0xcaca458fU, 0x82829d1fU, 0xc9c94089U, 0x7d7d87faU,
    0xfafa15efU, 0x5959ebb2U, 0x4747c98eU, 0xf0f00bfbU,
    0xadadec41U, 0xd4d467b3U, 0xa2a2fd5fU, 0xafafea45U,
    0x9c9cbf23U, 0xa4a4f753U, 0x727296e4U, 0xc0c05b9bU,
    0xb7b7c275U, 0xfdfd1ce1U, 0x9393ae3dU, 0x26266a4cU,
    0x36365a6cU, 0x3f3f417eU, 0xf7f702f5U, 0xcccc4f83U,
    0x34345c68U, 0xa5a5f451U, 0xe5e534d1U, 0xf1f108f9U,
    0x717193e2U, 0xd8d873abU, 0x31315362U, 0x15153f2aU,
    0x04040c08U, 0xc7c75295U, 0x23236546U, 0xc3c35e9dU,
    0x18182830U, 0x9696a137U, 0x05050f0aU, 0x9a9ab52fU,
    0x0707090eU, 0x12123624U, 0x80809b1bU, 0xe2e23ddfU,
    0xebeb26cdU, 0x2727694eU, 0xb2b2cd7fU, 0x75759feaU,
    0x09091b12U, 0x83839e1dU, 0x2c2c7458U, 0x1a1a2e34U,
    0x1b1b2d36U, 0x6e6eb2dcU, 0x5a5aeeb4U, 0xa0a0fb5bU,
    0x5252f6a4U, 0x3b3b4d76U, 0xd6d661b7U, 0xb3b3ce7dU,
    0x29297b52U, 0xe3e33eddU, 0x2f2f715eU, 0x84849713U,
    0x5353f5a6U, 0xd1d168b9U, 0x00000000U, 0xeded2cc1U,
    0x20206040U, 0xfcfc1fe3U, 0xb1b1c879U, 0x5b5bedb6U,
    0x6a6abed4U, 0xcbcb468dU, 0xbebed967U, 0x39394b72U,
    0x4a4ade94U, 0x4c4cd498U, 0x5858e8b0U, 0xcfcf4a85U,
    0xd0d06bbbU, 0xefef2ac5U, 0xaaaae54fU, 0xfbfb16edU,
    0x4343c586U, 0x4d4dd79aU, 0x33335566U, 0x85859411U,
    0x4545cf8aU, 0xf9f910e9U, 0x02020604U, 0x7f7f81feU,
    0x5050f0a0U, 0x3c3c4478U, 0x9f9fba25U, 0xa8a8e34bU,
    0x5151f3a2U, 0xa3a3fe5dU, 0x4040c080U, 0x8f8f8a05U,
    0x9292ad3fU, 0x9d9dbc21U, 0x38384870U, 0xf5f504f1U,
    0xbcbcdf63U, 0xb6b6c177U, 0xdada75afU, 0x21216342U,
    0x10103020U, 0xffff1ae5U, 0xf3f30efdU, 0xd2d26dbfU,
    0xcdcd4c81U, 0x0c0c1418U, 0x13133526U, 0xecec2fc3U,
    0x5f5fe1beU, 0x9797a235U, 0x4444cc88U, 0x1717392eU,
    0xc4c45793U, 0xa7a7f255U, 0x7e7e82fcU, 0x3d3d477aU,
    0x6464acc8U, 0x5d5de7baU, 0x19192b32U, 0x737395e6U,
    0x6060a0c0U, 0x81819819U, 0x4f4fd19eU, 0xdcdc7fa3U,
    0x22226644U, 0x2a2a7e54U, 0x9090ab3bU, 0x8888830bU,
    0x4646ca8cU, 0xeeee29c7U, 0xb8b8d36bU, 0x14143c28U,
    0xdede79a7U, 0x5e5ee2bcU, 0x0b0b1d16U, 0xdbdb76adU,
    0xe0e03bdbU, 0x32325664U, 0x3a3a4e74U, 0x0a0a1e14U,
    0x4949db92U, 0x06060a0cU, 0x24246c48U, 0x5c5ce4b8U,
    0xc2c25d9fU, 0xd3d36ebdU, 0xacacef43U, 0x6262a6c4U,
    0x9191a839U, 0x9595a431U, 0xe4e437d3U, 0x79798bf2U,
    0xe7e732d5U, 0xc8c8438bU, 0x3737596eU, 0x6d6db7daU,
    0x8d8d8c01U, 0xd5d564b1U, 0x4e4ed29cU, 0xa9a9e049U,
    0x6c6cb4d8U, 0x5656faacU, 0xf4f407f3U, 0xeaea25cfU,
    0x6565afcaU, 0x7a7a8ef4U, 0xaeaee947U, 0x08081810U,
    0xbabad56fU, 0x787888f0U, 0x25256f4aU, 0x2e2e725cU,
    0x1c1c2438U, 0xa6a6f157U, 0xb4b4c773U, 0xc6c65197U,
    0xe8e823cbU, 0xdddd7ca1U, 0x74749ce8U, 0x1f1f213eU,
    0x4b4bdd96U, 0xbdbddc61U, 0x8b8b860dU, 0x8a8a850fU,
    0x707090e0U, 0x3e3e427cU, 0xb5b5c471U, 0x6666aaccU,
    0x4848d890U, 0x03030506U, 0xf6f601f7U, 0x0e0e121cU,
    0x6161a3c2U, 0x35355f6aU, 0x5757f9aeU, 0xb9b9d069U,
    0x86869117U, 0xc1c15899U, 0x1d1d273aU, 0x9e9eb927U,
    0xe1e138d9U, 0xf8f813ebU, 0x9898b32bU, 0x11113322U,
    0x6969bbd2U, 0xd9d970a9U, 0x8e8e8907U, 0x9494a733U,
    0x9b9bb62dU, 0x1e1e223cU, 0x87879215U, 0xe9e920c9U,
    0xcece4987U, 0x5555ffaaU, 0x28287850U, 0xdfdf7aa5U,
    0x8c8c8f03U, 0xa1a1f859U, 0x89898009U, 0x0d0d171aU,
    0xbfbfda65U, 0xe6e631d7U, 0x4242c684U, 0x6868b8d0U,
    0x4141c382U, 0x9999b029U, 0x2d2d775aU, 0x0f0f111eU,
    0xb0b0cb7bU, 0x5454fca8U, 0xbbbbd66dU, 0x16163a2cU,
};

aes_decrypt_table::table_type aes_decrypt_table::Td0 = {
    0x51f4a750U, 0x7e416553U, 0x1a17a4c3U, 0x3a275e96U,
    0x3bab6bcbU, 0x1f9d45f1U, 0xacfa58abU, 0x4be30393U,
    0x2030fa55U, 0xad766df6U, 0x88cc7691U, 0xf5024c25U,
    0x4fe5d7fcU, 0xc52acbd7U, 0x26354480U, 0xb562a38fU,
    0xdeb15a49U, 0x25ba1b67U, 0x45ea0e98U, 0x5dfec0e1U,
    0xc32f7502U, 0x814cf012U, 0x8d4697a3U, 0x6bd3f9c6U,
    0x038f5fe7U, 0x15929c95U, 0xbf6d7aebU, 0x955259daU,
    0xd4be832dU, 0x587421d3U, 0x49e06929U, 0x8ec9c844U,
    0x75c2896aU, 0xf48e7978U, 0x99583e6bU, 0x27b971ddU,
    0xbee14fb6U, 0xf088ad17U, 0xc920ac66U, 0x7dce3ab4U,
    0x63df4a18U, 0xe51a3182U, 0x97513360U, 0x62537f45U,
    0xb16477e0U, 0xbb6bae84U, 0xfe81a01cU, 0xf9082b94U,
    0x70486858U, 0x8f45fd19U, 0x94de6c87U, 0x527bf8b7U,
    0xab73d323U, 0x724b02e2U, 0xe31f8f57U, 0x6655ab2aU,
    0xb2eb2807U, 0x2fb5c203U, 0x86c57b9aU, 0xd33708a5U,
    0x302887f2U, 0x23bfa5b2U, 0x02036abaU, 0xed16825cU,
    0x8acf1c2bU, 0xa779b492U, 0xf307f2f0U, 0x4e69e2a1U,
    0x65daf4cdU, 0x0605bed5U, 0xd134621fU, 0xc4a6fe8aU,
    0x342e539dU, 0xa2f355a0U, 0x058ae132U, 0xa4f6eb75U,
    0x0b83ec39U, 0x4060efaaU, 0x5e719f06U, 0xbd6e1051U,
    0x3e218af9U, 0x96dd063dU, 0xdd3e05aeU, 0x4de6bd46U,
    0x91548db5U, 0x71c45d05U, 0x0406d46fU, 0x605015ffU,
    0x1998fb24U, 0xd6bde997U, 0x894043ccU, 0x67d99e77U,
    0xb0e842bdU, 0x07898b88U, 0xe7195b38U, 0x79c8eedbU,
    0xa17c0a47U, 0x7c420fe9U, 0xf8841ec9U, 0x00000000U,
    0x09808683U, 0x322bed48U, 0x1e1170acU, 0x6c5a724eU,
    0xfd0efffbU, 0x0f853856U, 0x3daed51eU, 0x362d3927U,
    0x0a0fd964U, 0x685ca621U, 0x9b5b54d1U, 0x24362e3aU,
    0x0c0a67b1U, 0x9357e70fU, 0xb4ee96d2U, 0x1b9b919eU,
    0x80c0c54fU, 0x61dc20a2U, 0x5a774b69U, 0x1c121a16U,
    0xe293ba0aU, 0xc0a02ae5U, 0x3c22e043U, 0x121b171dU,
    0x0e090d0bU, 0xf28bc7adU, 0x2db6a8b9U, 0x141ea9c8U,
    0x57f11985U, 0xaf75074cU, 0xee99ddbbU, 0xa37f60fdU,
    0xf701269fU, 0x5c72f5bcU, 0x44663bc5U, 0x5bfb7e34U,
    0x8b432976U, 0xcb23c6dcU, 0xb6edfc68U, 0xb8e4f163U,
    0xd731dccaU, 0x42638510U, 0x13972240U, 0x84c61120U,
    0x854a247dU, 0xd2bb3df8U, 0xaef93211U, 0xc729a16dU,
    0x1d9e2f4bU, 0xdcb230f3U, 0x0d8652ecU, 0x77c1e3d0U,
    0x2bb3166cU, 0xa970b999U, 0x119448faU, 0x47e96422U,
    0xa8fc8cc4U, 0xa0f03f1aU, 0x567d2cd8U, 0x223390efU,
    0x87494ec7U, 0xd938d1c1U, 0x8ccaa2feU, 0x98d40b36U,
    0xa6f581cfU, 0xa57ade28U, 0xdab78e26U, 0x3fadbfa4U,
    0x2c3a9de4U, 0x5078920dU, 0x6a5fcc9bU, 0x547e4662U,
    0xf68d13c2U, 0x90d8b8e8U, 0x2e39f75eU, 0x82c3aff5U,
    0x9f5d80beU, 0x69d0937cU, 0x6fd52da9U, 0xcf2512b3U,
    0xc8ac993bU, 0x10187da7U, 0xe89c636eU, 0xdb3bbb7bU,
    0xcd267809U, 0x6e5918f4U, 0xec9ab701U, 0x834f9aa8U,
    0xe6956e65U, 0xaaffe67eU, 0x21bccf08U, 0xef15e8e6U,
    0xbae79bd9U, 0x4a6f36ceU, 0xea9f09d4U, 0x29b07cd6U,
    0x31a4b2afU, 0x2a3f2331U, 0xc6a59430U, 0x35a266c0U,
    0x744ebc37U, 0xfc82caa6U, 0xe090d0b0U, 0x33a7d815U,
    0xf104984aU, 0x41ecdaf7U, 0x7fcd500eU, 0x1791f62fU,
    0x764dd68dU, 0x43efb04dU, 0xccaa4d54U, 0xe49604dfU,
    0x9ed1b5e3U, 0x4c6a881bU, 0xc12c1fb8U, 0x4665517fU,
    0x9d5eea04U, 0x018c355dU, 0xfa877473U, 0xfb0b412eU,
    0xb3671d5aU, 0x92dbd252U, 0xe9105633U, 0x6dd64713U,
    0x9ad7618cU, 0x37a10c7aU, 0x59f8148eU, 0xeb133c89U,
    0xcea927eeU, 0xb761c935U, 0xe11ce5edU, 0x7a47b13cU,
    0x9cd2df59U, 0x55f2733fU, 0x1814ce79U, 0x73c737bfU,
    0x53f7cdeaU, 0x5ffdaa5bU, 0xdf3d6f14U, 0x7844db86U,
    0xcaaff381U, 0xb968c43eU, 0x3824342cU, 0xc2a3405fU,
    0x161dc372U, 0xbce2250cU, 0x283c498bU, 0xff0d9541U,
    0x39a80171U, 0x080cb3deU, 0xd8b4e49cU, 0x6456c190U,
    0x7bcb8461U, 0xd532b670U, 0x486c5c74U, 0xd0b85742U,
};

aes_decrypt_table::table_type aes_decrypt_table::Td1 = {
    0x5051f4a7U, 0x537e4165U, 0xc31a17a4U, 0x963a275eU,
    0xcb3bab6bU, 0xf11f9d45U, 0xabacfa58U, 0x934be303U,
    0x552030faU, 0xf6ad766dU, 0x9188cc76U, 0x25f5024cU,
    0xfc4fe5d7U, 0xd7c52acbU, 0x80263544U, 0x8fb562a3U,
    0x49deb15aU, 0x6725ba1bU, 0x9845ea0eU, 0xe15dfec0U,
    0x02c32f75U, 0x12814cf0U, 0xa38d4697U, 0xc66bd3f9U,
    0xe7038f5fU, 0x9515929cU, 0xebbf6d7aU, 0xda955259U,
    0x2dd4be83U, 0xd3587421U, 0x2949e069U, 0x448ec9c8U,
    0x6a75c289U, 0x78f48e79U, 0x6b99583eU, 0xdd27b971U,
    0xb6bee14fU, 0x17f088adU, 0x66c920acU, 0xb47dce3aU,
    0x1863df4aU, 0x82e51a31U, 0x60975133U, 0x4562537fU,
    0xe0b16477U, 0x84bb6baeU, 0x1cfe81a0U, 0x94f9082bU,
    0x58704868U, 0x198f45fdU, 0x8794de6cU, 0xb7527bf8U,
    0x23ab73d3U, 0xe2724b02U, 0x57e31f8fU, 0x2a6655abU,
    0x07b2eb28U, 0x032fb5c2U, 0x9a86c57bU, 0xa5d33708U,
    0xf2302887U, 0xb223bfa5U, 0xba02036aU, 0x5ced1682U,
    0x2b8acf1cU, 0x92a779b4U, 0xf0f307f2U, 0xa14e69e2U,
    0xcd65daf4U, 0xd50605beU, 0x1fd13462U, 0x8ac4a6feU,
    0x9d342e53U, 0xa0a2f355U, 0x32058ae1U, 0x75a4f6ebU,
    0x390b83ecU, 0xaa4060efU, 0x065e719fU, 0x51bd6e10U,
    0xf93e218aU, 0x3d96dd06U, 0xaedd3e05U, 0x464de6bdU,
    0xb591548dU, 0x0571c45dU, 0x6f0406d4U, 0xff605015U,
    0x241998fbU, 0x97d6bde9U, 0xcc894043U, 0x7767d99eU,
    0xbdb0e842U, 0x8807898bU, 0x38e7195bU, 0xdb79c8eeU,
    0x47a17c0aU, 0xe97c420fU, 0xc9f8841eU, 0x00000000U,
    0x83098086U, 0x48322bedU, 0xac1e1170U, 0x4e6c5a72U,
    0xfbfd0effU, 0x560f8538U, 0x1e3daed5U, 0x27362d39U,
    0x640a0fd9U, 0x21685ca6U, 0xd19b5b54U, 0x3a24362eU,
    0xb10c0a67U, 0x0f9357e7U, 0xd2b4ee96U, 0x9e1b9b91U,
    0x4f80c0c5U, 0xa261dc20U, 0x695a774bU, 0x161c121aU,
    0x0ae293baU, 0xe5c0a02aU, 0x433c22e0U, 0x1d121b17U,
    0x0b0e090dU, 0xadf28bc7U, 0xb92db6a8U, 0xc8141ea9U,
    0x8557f119U, 0x4caf7507U, 0xbbee99ddU, 0xfda37f60U,
    0x9ff70126U, 0xbc5c72f5U, 0xc544663bU, 0x345bfb7eU,
    0x768b4329U, 0xdccb23c6U, 0x68b6edfcU, 0x63b8e4f1U,
    0xcad731dcU, 0x10426385U, 0x40139722U, 0x2084c611U,
    0x7d854a24U, 0xf8d2bb3dU, 0x11aef932U, 0x6dc729a1U,
    0x4b1d9e2fU, 0xf3dcb230U, 0xec0d8652U, 0xd077c1e3U,
    0x6c2bb316U, 0x99a970b9U, 0xfa119448U, 0x2247e964U,
    0xc4a8fc8cU, 0x1aa0f03fU, 0xd8567d2cU, 0xef223390U,
    0xc787494eU, 0xc1d938d1U, 0xfe8ccaa2U, 0x3698d40bU,
    0xcfa6f581U, 0x28a57adeU, 0x26dab78eU, 0xa43fadbfU,
    0xe42c3a9dU, 0x0d507892U, 0x9b6a5fccU, 0x62547e46U,
    0xc2f68d13U, 0xe890d8b8U, 0x5e2e39f7U, 0xf582c3afU,
    0xbe9f5d80U, 0x7c69d093U, 0xa96fd52dU, 0xb3cf2512U,
    0x3bc8ac99U, 0xa710187dU, 0x6ee89c63U, 0x7bdb3bbbU,
    0x09cd2678U, 0xf46e5918U, 0x01ec9ab7U, 0xa8834f9aU,
    0x65e6956eU, 0x7eaaffe6U, 0x0821bccfU, 0xe6ef15e8U,
    0xd9bae79bU, 0xce4a6f36U, 0xd4ea9f09U, 0xd629b07cU,
    0xaf31a4b2U, 0x312a3f23U, 0x30c6a594U, 0xc035a266U,
    0x37744ebcU, 0xa6fc82caU, 0xb0e090d0U, 0x1533a7d8U,
    0x4af10498U, 0xf741ecdaU, 0x0e7fcd50U, 0x2f1791f6U,
    0x8d764dd6U, 0x4d43efb0U, 0x54ccaa4dU, 0xdfe49604U,
    0xe39ed1b5U, 0x1b4c6a88U, 0xb8c12c1fU, 0x7f466551U,
    0x049d5eeaU, 0x5d018c35U, 0x73fa8774U, 0x2efb0b41U,
    0x5ab3671dU, 0x5292dbd2U, 0x33e91056U, 0x136dd647U,
    0x8c9ad761U, 0x7a37a10cU, 0x8e59f814U, 0x89eb133cU,
    0xeecea927U, 0x35b761c9U, 0xede11ce5U, 0x3c7a47b1U,
    0x599cd2dfU, 0x3f55f273U, 0x791814ceU, 0xbf73c737U,
    0xea53f7cdU, 0x5b5ffdaaU, 0x14df3d6fU, 0x867844dbU,
    0x81caaff3U, 0x3eb968c4U, 0x2c382434U, 0x5fc2a340U,
    0x72161dc3U, 0x0cbce225U, 0x8b283c49U, 0x41ff0d95U,
    0x7139a801U, 0xde080cb3U, 0x9cd8b4e4U, 0x906456c1U,
    0x617bcb84U, 0x70d532b6U, 0x74486c5cU, 0x42d0b857U,
};

aes_decrypt_table::table_type aes_decrypt_table::Td2 = {
    0xa75051f4U, 0x65537e41U, 0xa4c31a17U, 0x5e963a27U,
    0x6bcb3babU, 0x45f11f9dU, 0x58abacfaU, 0x03934be3U,
    0xfa552030U, 0x6df6ad76U, 0x769188ccU, 0x4c25f502U,
    0xd7fc4fe5U, 0xcbd7c52aU, 0x44802635U, 0xa38fb562U,
    0x5a49deb1U, 0x1b6725baU, 0x0e9845eaU, 0xc0e15dfeU,
    0x7502c32fU, 0xf012814cU, 0x97a38d46U, 0xf9c66bd3U,
    0x5fe7038fU, 0x9c951592U, 0x7aebbf6dU, 0x59da9552U,
    0x832dd4beU, 0x21d35874U, 0x692949e0U, 0xc8448ec9U,
    0x896a75c2U, 0x7978f48eU, 0x3e6b9958U, 0x71dd27b9U,
    0x4fb6bee1U, 0xad17f088U, 0xac66c920U, 0x3ab47dceU,
    0x4a1863dfU, 0x3182e51aU, 0x33609751U, 0x7f456253U,
    0x77e0b164U, 0xae84bb6bU, 0xa01cfe81U, 0x2b94f908U,
    0x68587048U, 0xfd198f45U, 0x6c8794deU, 0xf8b7527bU,
    0xd323ab73U, 0x02e2724bU, 0x8f57e31fU, 0xab2a6655U,
    0x2807b2ebU, 0xc2032fb5U, 0x7b9a86c5U, 0x08a5d337U,
    0x87f23028U, 0xa5b223bfU, 0x6aba0203U, 0x825ced16U,
    0x1c2b8acfU, 0xb492a779U, 0xf2f0f307U, 0xe2a14e69U,
    0xf4cd65daU, 0xbed50605U, 0x621fd134U, 0xfe8ac4a6U,
    0x539d342eU, 0x55a0a2f3U, 0xe132058aU, 0xeb75a4f6U,
    0xec390b83U, 0xefaa4060U, 0x9f065e71U, 0x1051bd6eU,
    0x8af93e21U, 0x063d96ddU, 0x05aedd3eU, 0xbd464de6U,
    0x8db59154U, 0x5d0571c4U, 0xd46f0406U, 0x15ff6050U,
    0xfb241998U, 0xe997d6bdU, 0x43cc8940U, 0x9e7767d9U,
    0x42bdb0e8U, 0x8b880789U, 0x5b38e719U, 0xeedb79c8U,
    0x0a47a17cU, 0x0fe97c42U, 0x1ec9f884U, 0x00000000U,
    0x86830980U, 0xed48322bU, 0x70ac1e11U, 0x724e6c5aU,
    0xfffbfd0eU, 0x38560f85U, 0xd51e3daeU, 0x3927362dU,
    0xd9640a0fU, 0xa621685cU, 0x54d19b5bU, 0x2e3a2436U,
    0x67b10c0aU, 0xe70f9357U, 0x96d2b4eeU, 0x919e1b9bU,
    0xc54f80c0U, 0x20a261dcU, 0x4b695a77U, 0x1a161c12U,
    0xba0ae293U, 0x2ae5c0a0U, 0xe0433c22U, 0x171d121bU,
    0x0d0b0e09U, 0xc7adf28bU, 0xa8b92db6U, 0xa9c8141eU,
    0x198557f1U, 0x074caf75U, 0xddbbee99U, 0x60fda37fU,
    0x269ff701U, 0xf5bc5c72U, 0x3bc54466U, 0x7e345bfbU,
    0x29768b43U, 0xc6dccb23U, 0xfc68b6edU, 0xf163b8e4U,
    0xdccad731U, 0x85104263U, 0x22401397U, 0x112084c6U,
    0x247d854aU, 0x3df8d2bbU, 0x3211aef9U, 0xa16dc729U,
    0x2f4b1d9eU, 0x30f3dcb2U, 0x52ec0d86U, 0xe3d077c1U,
    0x166c2bb3U, 0xb999a970U, 0x48fa1194U, 0x642247e9U,
    0x8cc4a8fcU, 0x3f1aa0f0U, 0x2cd8567dU, 0x90ef2233U,
    0x4ec78749U, 0xd1c1d938U, 0xa2fe8ccaU, 0x0b3698d4U,
    0x81cfa6f5U, 0xde28a57aU, 0x8e26dab7U, 0xbfa43fadU,
    0x9de42c3aU, 0x920d5078U, 0xcc9b6a5fU, 0x4662547eU,
    0x13c2f68dU, 0xb8e890d8U, 0xf75e2e39U, 0xaff582c3U,
    0x80be9f5dU, 0x937c69d0U, 0x2da96fd5U, 0x12b3cf25U,
    0x993bc8acU, 0x7da71018U, 0x636ee89cU, 0xbb7bdb3bU,
    0x7809cd26U, 0x18f46e59U, 0xb701ec9aU, 0x9aa8834fU,
    0x6e65e695U, 0xe67eaaffU, 0xcf0821bcU, 0xe8e6ef15U,
    0x9bd9bae7U, 0x36ce4a6fU, 0x09d4ea9fU, 0x7cd629b0U,
    0xb2af31a4U, 0x23312a3fU, 0x9430c6a5U, 0x66c035a2U,
    0xbc37744eU, 0xcaa6fc82U, 0xd0b0e090U, 0xd81533a7U,
    0x984af104U, 0xdaf741ecU, 0x500e7fcdU, 0xf62f1791U,
    0xd68d764dU, 0xb04d43efU, 0x4d54ccaaU, 0x04dfe496U,
    0xb5e39ed1U, 0x881b4c6aU, 0x1fb8c12cU, 0x517f4665U,
    0xea049d5eU, 0x355d018cU, 0x7473fa87U, 0x412efb0bU,
    0x1d5ab367U, 0xd25292dbU, 0x5633e910U, 0x47136dd6U,
    0x618c9ad7U, 0x0c7a37a1U, 0x148e59f8U, 0x3c89eb13U,
    0x27eecea9U, 0xc935b761U, 0xe5ede11cU, 0xb13c7a47U,
    0xdf599cd2U, 0x733f55f2U, 0xce791814U, 0x37bf73c7U,
    0xcdea53f7U, 0xaa5b5ffdU, 0x6f14df3dU, 0xdb867844U,
    0xf381caafU, 0xc43eb968U, 0x342c3824U, 0x405fc2a3U,
    0xc372161dU, 0x250cbce2U, 0x498b283cU, 0x9541ff0dU,
    0x017139a8U, 0xb3de080cU, 0xe49cd8b4U, 0xc1906456U,
    0x84617bcbU, 0xb670d532U, 0x5c74486cU, 0x5742d0b8U,
};

aes_decrypt_table::table_type aes_decrypt_table::Td3 = {
    0xf4a75051U, 0x4165537eU, 0x17a4c31aU, 0x275e963aU,
    0xab6bcb3bU, 0x9d45f11fU, 0xfa58abacU, 0xe303934bU,
    0x30fa5520U, 0x766df6adU, 0xcc769188U, 0x024c25f5U,
    0xe5d7fc4fU, 0x2acbd7c5U, 0x35448026U, 0x62a38fb5U,
    0xb15a49deU, 0xba1b6725U, 0xea0e9845U, 0xfec0e15dU,
    0x2f7502c3U, 0x4cf01281U, 0x4697a38dU, 0xd3f9c66bU,
    0x8f5fe703U, 0x929c9515U, 0x6d7aebbfU, 0x5259da95U,
    0xbe832dd4U, 0x7421d358U, 0xe0692949U, 0xc9c8448eU,
    0xc2896a75U, 0x8e7978f4U, 0x583e6b99U, 0xb971dd27U,
    0xe14fb6beU, 0x88ad17f0U, 0x20ac66c9U, 0xce3ab47dU,
    0xdf4a1863U, 0x1a3182e5U, 0x51336097U, 0x537f4562U,
    0x6477e0b1U, 0x6bae84bbU, 0x81a01cfeU, 0x082b94f9U,
    0x48685870U, 0x45fd198fU, 0xde6c8794U, 0x7bf8b752U,
    0x73d323abU, 0x4b02e272U, 0x1f8f57e3U, 0x55ab2a66U,
    0xeb2807b2U, 0xb5c2032fU, 0xc57b9a86U, 0x3708a5d3U,
    0x2887f230U, 0xbfa5b223U, 0x036aba02U, 0x16825cedU,
    0xcf1c2b8aU, 0x79b492a7U, 0x07f2f0f3U, 0x69e2a14eU,
    0xdaf4cd65U, 0x05bed506U, 0x34621fd1U, 0xa6fe8ac4U,
    0x2e539d34U, 0xf355a0a2U, 0x8ae13205U, 0xf6eb75a4U,
    0x83ec390bU, 0x60efaa40U, 0x719f065eU, 0x6e1051bdU,
    0x218af93eU, 0xdd063d96U, 0x3e05aeddU, 0xe6bd464dU,
    0x548db591U, 0xc45d0571U, 0x06d46f04U, 0x5015ff60U,
    0x98fb2419U, 0xbde997d6U, 0x4043cc89U, 0xd99e7767U,
    0xe842bdb0U, 0x898b8807U, 0x195b38e7U, 0xc8eedb79U,
    0x7c0a47a1U, 0x420fe97cU, 0x841ec9f8U, 0x00000000U,
    0x80868309U, 0x2bed4832U, 0x1170ac1eU, 0x5a724e6cU,
    0x0efffbfdU, 0x8538560fU, 0xaed51e3dU, 0x2d392736U,
    0x0fd9640aU, 0x5ca62168U, 0x5b54d19bU, 0x362e3a24U,
    0x0a67b10cU, 0x57e70f93U, 0xee96d2b4U, 0x9b919e1bU,
    0xc0c54f80U, 0xdc20a261U, 0x774b695aU, 0x121a161cU,
    0x93ba0ae2U, 0xa02ae5c0U, 0x22e0433cU, 0x1b171d12U,
    0x090d0b0eU, 0x8bc7adf2U, 0xb6a8b92dU, 0x1ea9c814U,
    0xf1198557U, 0x75074cafU, 0x99ddbbeeU, 0x7f60fda3U,
    0x01269ff7U, 0x72f5bc5cU, 0x663bc544U, 0xfb7e345bU,
    0x4329768bU, 0x23c6dccbU, 0xedfc68b6U, 0xe4f163b8U,
    0x31dccad7U, 0x63851042U, 0x97224013U, 0xc6112084U,
    0x4a247d85U, 0xbb3df8d2U, 0xf93211aeU, 0x29a16dc7U,
    0x9e2f4b1dU, 0xb230f3dcU, 0x8652ec0dU, 0xc1e3d077U,
    0xb3166c2bU, 0x70b999a9U, 0x9448fa11U, 0xe9642247U,
    0xfc8cc4a8U, 0xf03f1aa0U, 0x7d2cd856U, 0x3390ef22U,
    0x494ec787U, 0x38d1c1d9U, 0xcaa2fe8cU, 0xd40b3698U,
    0xf581cfa6U, 0x7ade28a5U, 0xb78e26daU, 0xadbfa43fU,
    0x3a9de42cU, 0x78920d50U, 0x5fcc9b6aU, 0x7e466254U,
    0x8d13c2f6U, 0xd8b8e890U, 0x39f75e2eU, 0xc3aff582U,
    0x5d80be9fU, 0xd0937c69U, 0xd52da96fU, 0x2512b3cfU,
    0xac993bc8U, 0x187da710U, 0x9c636ee8U, 0x3bbb7bdbU,
    0x267809cdU, 0x5918f46eU, 0x9ab701ecU, 0x4f9aa883U,
    0x956e65e6U, 0xffe67eaaU, 0xbccf0821U, 0x15e8e6efU,
    0xe79bd9baU, 0x6f36ce4aU, 0x9f09d4eaU, 0xb07cd629U,
    0xa4b2af31U, 0x3f23312aU, 0xa59430c6U, 0xa266c035U,
    0x4ebc3774U, 0x82caa6fcU, 0x90d0b0e0U, 0xa7d81533U,
    0x04984af1U, 0xecdaf741U, 0xcd500e7fU, 0x91f62f17U,
    0x4dd68d76U, 0xefb04d43U, 0xaa4d54ccU, 0x9604dfe4U,
    0xd1b5e39eU, 0x6a881b4cU, 0x2c1fb8c1U, 0x65517f46U,
    0x5eea049dU, 0x8c355d01U, 0x877473faU, 0x0b412efbU,
    0x671d5ab3U, 0xdbd25292U, 0x105633e9U, 0xd647136dU,
    0xd7618c9aU, 0xa10c7a37U, 0xf8148e59U, 0x133c89ebU,
    0xa927eeceU, 0x61c935b7U, 0x1ce5ede1U, 0x47b13c7aU,
    0xd2df599cU, 0xf2733f55U, 0x14ce7918U, 0xc737bf73U,
    0xf7cdea53U, 0xfdaa5b5fU, 0x3d6f14dfU, 0x44db8678U,
    0xaff381caU, 0x68c43eb9U, 0x24342c38U, 0xa3405fc2U,
    0x1dc37216U, 0xe2250cbcU, 0x3c498b28U, 0x0d9541ffU,
    0xa8017139U, 0x0cb3de08U, 0xb4e49cd8U, 0x56c19064U,
    0xcb84617bU, 0x32b670d5U, 0x6c5c7448U, 0xb85742d0U,
};

const unsigned char aes_decrypt_table::Td4[table_nelts] = {
    0x52U, 0x09U, 0x6aU, 0xd5U, 0x30U, 0x36U, 0xa5U, 0x38U,
    0xbfU, 0x40U, 0xa3U, 0x9eU, 0x81U, 0xf3U, 0xd7U, 0xfbU,
    0x7cU, 0xe3U, 0x39U, 0x82U, 0x9bU, 0x2fU, 0xffU, 0x87U,
    0x34U, 0x8eU, 0x43U, 0x44U, 0xc4U, 0xdeU, 0xe9U, 0xcbU,
    0x54U, 0x7bU, 0x94U, 0x32U, 0xa6U, 0xc2U, 0x23U, 0x3dU,
    0xeeU, 0x4cU, 0x95U, 0x0bU, 0x42U, 0xfaU, 0xc3U, 0x4eU,
    0x08U, 0x2eU, 0xa1U, 0x66U, 0x28U, 0xd9U, 0x24U, 0xb2U,
    0x76U, 0x5bU, 0xa2U, 0x49U, 0x6dU, 0x8bU, 0xd1U, 0x25U,
    0x72U, 0xf8U, 0xf6U, 0x64U, 0x86U, 0x68U, 0x98U, 0x16U,
    0xd4U, 0xa4U, 0x5cU, 0xccU, 0x5dU, 0x65U, 0xb6U, 0x92U,
    0x6cU, 0x70U, 0x48U, 0x50U, 0xfdU, 0xedU, 0xb9U, 0xdaU,
    0x5eU, 0x15U, 0x46U, 0x57U, 0xa7U, 0x8dU, 0x9dU, 0x84U,
    0x90U, 0xd8U, 0xabU, 0x00U, 0x8cU, 0xbcU, 0xd3U, 0x0aU,
    0xf7U, 0xe4U, 0x58U, 0x05U, 0xb8U, 0xb3U, 0x45U, 0x06U,
    0xd0U, 0x2cU, 0x1eU, 0x8fU, 0xcaU, 0x3fU, 0x0fU, 0x02U,
    0xc1U, 0xafU, 0xbdU, 0x03U, 0x01U, 0x13U, 0x8aU, 0x6bU,
    0x3aU, 0x91U, 0x11U, 0x41U, 0x4fU, 0x67U, 0xdcU, 0xeaU,
    0x97U, 0xf2U, 0xcfU, 0xceU, 0xf0U, 0xb4U, 0xe6U, 0x73U,
    0x96U, 0xacU, 0x74U, 0x22U, 0xe7U, 0xadU, 0x35U, 0x85U,
    0xe2U, 0xf9U, 0x37U, 0xe8U, 0x1cU, 0x75U, 0xdfU, 0x6eU,
    0x47U, 0xf1U, 0x1aU, 0x71U, 0x1dU, 0x29U, 0xc5U, 0x89U,
    0x6fU, 0xb7U, 0x62U, 0x0eU, 0xaaU, 0x18U, 0xbeU, 0x1bU,
    0xfcU, 0x56U, 0x3eU, 0x4bU, 0xc6U, 0xd2U, 0x79U, 0x20U,
    0x9aU, 0xdbU, 0xc0U, 0xfeU, 0x78U, 0xcdU, 0x5aU, 0xf4U,
    0x1fU, 0xddU, 0xa8U, 0x33U, 0x88U, 0x07U, 0xc7U, 0x31U,
    0xb1U, 0x12U, 0x10U, 0x59U, 0x27U, 0x80U, 0xecU, 0x5fU,
    0x60U, 0x51U, 0x7fU, 0xa9U, 0x19U, 0xb5U, 0x4aU, 0x0dU,
    0x2dU, 0xe5U, 0x7aU, 0x9fU, 0x93U, 0xc9U, 0x9cU, 0xefU,
    0xa0U, 0xe0U, 0x3bU, 0x4dU, 0xaeU, 0x2aU, 0xf5U, 0xb0U,
    0xc8U, 0xebU, 0xbbU, 0x3cU, 0x83U, 0x53U, 0x99U, 0x61U,
    0x17U, 0x2bU, 0x04U, 0x7eU, 0xbaU, 0x77U, 0xd6U, 0x26U,
    0xe1U, 0x69U, 0x14U, 0x63U, 0x55U, 0x21U, 0x0cU, 0x7dU,
};

/* In-round shifts info.  */
static const unsigned HOST_WIDE_INT shift_csts[4] = {24, 16, 8, 0};

/* Check if the pattern is plus-const.  Helper for memref analysis.  */
static bool
plus_const_int_p (rtx op)
{
  return GET_CODE (op) == PLUS && CONST_INT_P (XEXP (op, 1));
}

/* Obtain info about memory access.  */
static bool
decompose_mem (rtx mem, rtx &base, unsigned HOST_WIDE_INT &offset)
{
  address_info info;
  decompose_mem_address (&info, mem);
  if (!info.base)
    return false;

  base = *info.base;

  rtx op = XEXP (mem, 0);
  if (plus_const_int_p (op))
    offset = UINTVAL (XEXP (op, 1));
  /* TODO: WRONG IN GENERAL CASE: we cannot guarantee that the offsets were not
     changed.  */
  else if ((GET_CODE (op) == PRE_MODIFY && plus_const_int_p (XEXP (op, 1)))
	   || REG_P (op))
    offset = 0;
  else
    return false;

  return true;
}

/* Check if the regs in stmt are same as the provided ones.  */
static bool
cmp_regs_in_stmt (rtx stmt, rtx lhs, rtx rhs)
{
  return (XEXP (stmt, 0) == lhs) && (XEXP (stmt, 1) == rhs);
}

/* AES key info.  Inhereted from mem_term_info to be used inside
   matchers without any unnecessary casts.  */
struct aes_key : mem_term_info
{
  aes_key ()
    {}
  aes_key (void *)
    : mem_term_info (NULL, NULL_RTX)
    {}
  aes_key (const mem_term_info &m)
    : mem_term_info (m)
    {}

  /* Check if the key has the same base pointer origin as another one.
     This check is required due to some possible CSE optimizations applied on
     pointers before this pass.  */
  bool has_same_origin (const aes_key &other, rtx_insn *use_point) const
    {
      /* Simple case: the pointer is same.  */
      if (src == other.src)
	return true;

      if (!use_point)
	return false;

      basic_block curr_bb = BLOCK_FOR_INSN (use_point);
      if (!single_pred_p (curr_bb)
	  || modified_between_p (src, BB_HEAD (curr_bb), use_point)
	  || modified_between_p (other.src, BB_HEAD (curr_bb), use_point))
	return false;

      edge e = single_pred_edge (curr_bb);
      rtx_insn *jump = BB_END (e->src);
      if (!any_condjump_p (jump))
	return false;

      basic_block from_bb = BLOCK_FOR_INSN (jump);
      if (EDGE_COUNT (from_bb->succs) != 2)
	return false;

      /* Need proof that the sources are equal: try to get it from
	 terminating condition.  */
      rtx cond = XEXP (SET_SRC (pc_set (jump)), 0);
      rtx_code code = GET_CODE (cond);
      if (!((code == EQ && EDGE_SUCC (from_bb, 0) == e)
	    || (code == NE && EDGE_SUCC (from_bb, 1) == e)))
	return false;

      rtx arg1 = XEXP (cond, 0);
      if (XEXP (cond, 1) != CONST0_RTX (GET_MODE (arg1))
	  || COMPARISON_P (arg1))
	return false;

      rtx_insn *cmp_insn = get_single_def_insn (jump, arg1);
      rtx cmp;
      if (!cmp_insn || !(cmp = get_single_set_op<COMPARE> (cmp_insn)))
	return false;

      if (!(cmp_regs_in_stmt (cmp, src, other.src)
	    || cmp_regs_in_stmt (cmp, other.src, src)))
	return false;

      return true;
    }
};

/* AES basic state input info.  Inhereted from mem_term_info
   to use it in matchers without any unnecessary casts.  */
struct state_input_info : mem_term_info
{
  state_input_info ()
    {}
  state_input_info (const aes_key &k)
    : mem_term_info (k), is_key (true)
    {}
  state_input_info (const mem_term_info &m)
    : mem_term_info (m), is_key (false)
    {}

  bool is_key;

  bool verify (const state_input_info *prev) const
    {
      if (!prev)
	return true;

      return BLOCK_FOR_INSN (loc) == BLOCK_FOR_INSN (prev->loc);
    }
};

/* Memory matcher to filter only suitable memory instructions.  */
struct mem_matcher : matcher_term<mem_term_info>
{
  static bool match (rtx_insn *insn, holder_type &m)
    {
      rtx src = get_single_set_op<MEM> (insn);
      return src && match (src, insn, m);
    }

  static bool match (rtx src, rtx_insn *insn, holder_type &m)
    {
      if (!MEM_P (src))
	return false;

      mem_term_info info (NULL, NULL_RTX);
      if (!decompose_mem (src, info.src, info.offset))
	return false;

      info.loc = insn;
      m[0] = info;
      return true;
    }
};

/* AES entry input info.  Enhanced from state input due to ideological
   similarities.  */
struct input_info : state_input_info
{
  input_info ()
    {}
  input_info (const mem_term_info &m, unsigned HOST_WIDE_INT shift_cst)
    : state_input_info (m), shift_cst (shift_cst)
    {}
  input_info (const aes_key &k)
    : state_input_info (k)
    {}

  unsigned HOST_WIDE_INT shift_cst;

  /* Input info is sorted by references offsets.  */
  bool operator < (const input_info &rhs) const
    {
      return offset < rhs.offset;
    }

  std::pair<rtx, unsigned HOST_WIDE_INT> input () const
    {
      return std::make_pair (src, offset);
    }

  bool verify (const input_info *prev, unsigned i) const
    {
      if (!state_input_info::verify (prev))
	return false;

      /* Previous state should reference the previous element
	 of the same buffer.  */
      if (prev && (src != prev->src || offset != prev->offset + 1))
	return false;

      /* State should use the corresponding shift constant.  */
      return shift_csts[i] == shift_cst;
    }

  static bool finalize (rtx_insn *insn, input_info *m)
    {
      typedef unop_matcher<ZERO_EXTEND, mem_matcher> zext_matcher;

      zext_matcher::holder_type zext;
      if (zext_matcher::match (insn, zext))
	{
	  *m = input_info (zext[0], 0);
	  return true;
	}

      typedef binop_matcher<ASHIFT,
			    zext_matcher, int_cst_matcher<mem_term_info> >
	shifted_variant;
      shifted_variant::holder_type lsh;
      if (!shifted_variant::match (insn, lsh))
	return false;

      gcc_assert (CONST_INT_P (lsh[1].src));
      *m = input_info (lsh[0], UINTVAL (lsh[1].src));
      return true;
    }
};

/* Check if the corresponding constants combinations may be used for
   AES table access.  */
static bool
verify_table_access (unsigned HOST_WIDE_INT shift_cst,
		     unsigned HOST_WIDE_INT and_cst = 0xFF,
		     bool and_present = true)
{
  if (and_cst != 0xFF)
    return false;

  switch (shift_cst)
    {
    case 0:
    case 8:
    case 16:
      return and_present;
    case 24:
      return true;
    default:
      return false;
    }
}

/* AES table reference description.  */
template<typename TABLE_T>
struct aes_table_ref
{
  rtx_insn *insn;
  rtx_insn *output_insn;
  unsigned HOST_WIDE_INT lsr_cst;
  rtx reg;
  rtx output;
  typename TABLE_T::table_entry itable;
  bool is_final;

  bool verify (unsigned i) const
    {
      typename TABLE_T::table_entry (&ethalon)[TABLE_T::rounds_num]
	= is_final ? TABLE_T::final_rounds : TABLE_T::rounds;
      return lsr_cst == shift_csts[i] && itable == ethalon[i];
    }
};

/* Check the minimal requirements of the pattern to be a table reference
   and wrap the table id getter function.  */
template<typename T>
static typename T::table_entry
check_table (rtx mem)
{
  tree expr = MEM_EXPR (mem);
  if (!expr || TREE_CODE (expr) != ARRAY_REF)
    return T::BAD_TABLE;

  tree decl = TREE_OPERAND (expr, 0);
  if (!decl || !DECL_P (decl) || !TREE_READONLY (decl))
    return T::BAD_TABLE;

  tree ctor = DECL_INITIAL (decl);
  if (!ctor)
    return T::BAD_TABLE;

  return T::get_table_id (ctor);
}

/* Simplified memory info.  Used for simplier table ref analysis.  */
struct simplified_mem_info
{
  rtx base_reg;
  rtx index;
};

/* Try to obtain table reference info.  */
static bool
decompose_tref_mem_address (simplified_mem_info &info, rtx mem)
{
  address_info addr_info;
  decompose_mem_address (&addr_info, mem);
  if (!addr_info.base || !addr_info.index)
    return false;

  info.base_reg = *addr_info.base;
  info.index = *addr_info.index;

  if (!REG_P (info.base_reg))
    return false;

  if (addr_info.mode == SImode)
    {
      if (GET_CODE (info.index) != MULT)
	return false;

      rtx cst = XEXP (info.index, 1);
      if (!CONST_INT_P (cst) || UINTVAL (cst) != 4)
	return false;

      info.index = XEXP (info.index, 0);
      return true;
    }

  return (addr_info.mode == QImode);
}

/* Find the possible final output instruction.  */
template<typename TABLE_T>
static rtx_insn *
get_possible_final_output (rtx_insn *insn, rtx reg,
			   unsigned HOST_WIDE_INT shift_cst,
			   typename TABLE_T::table_entry itable);

/* Specialize the function for AES encryption.  The output is AND instruction
   with propper constant.  */
template<>
rtx_insn *
get_possible_final_output<aes_encrypt_table> (rtx_insn *insn, rtx reg,
					      unsigned HOST_WIDE_INT shift_cst,
					      aes_encrypt_table::table_entry)
{
  rtx_insn *out = get_single_use_insn (insn, reg);
  if (!out)
    return NULL;

  rtx cst_val = get_op_const_cst<AND> (out);
  if (!cst_val)
    return NULL;

  unsigned HOST_WIDE_INT ethalon;
  switch (shift_cst)
    {
    case 24:
      ethalon = 0xffffffffff000000;
      break;
    case 16:
      ethalon = 0xff0000;
      break;
    case 8:
      ethalon = 0xff00;
      break;
    case 0:
      ethalon = 0xff;
      break;
    default:
      gcc_unreachable ();
    }

  return UINTVAL (cst_val) == ethalon ? out : NULL;
}

/* Specialize the function for AES decryption.  The output is ASHIFT instruction
   with propper constant or direct reference to TD4 table.

   TODO: TD4 check might be done here for all the cases.  However, now it is not
   done here to make decryption and encryption matching
   more general in common.  */
template<>
rtx_insn *
get_possible_final_output<aes_decrypt_table> (rtx_insn *insn, rtx reg,
					      unsigned HOST_WIDE_INT shift_cst,
					      aes_decrypt_table::table_entry it)
{
  rtx_insn *out = get_single_use_insn (insn, reg);
  if (!out)
    return NULL;

  rtx cst_val = get_op_const_cst<ASHIFT> (out);
  if (!cst_val)
    // no shift case
    return it == aes_decrypt_table::TD4 ? insn : NULL;

  return UINTVAL (cst_val) == shift_cst ? out : NULL;
}

typedef arg_op_matcher<REG> reg_matcher;

/* Helper that matches suitable AES table references.  */
template<typename TABLE_T>
class tref_matcher
{
  /* (reg >> cst) matcher.  Helper.  */
  typedef binop_matcher<LSHIFTRT,
			reg_matcher,
			int_cst_matcher<minimal_term_info> > table_access;
  /* zext (reg >> cst) matcher.  Used for TABLE[(val >> 24)] variant.  */
  typedef unop_matcher<ZERO_EXTEND, table_access> direct;
  /* zext ((reg >> cst1) & cst2) matcher.  Used for
     TABLE[(val >> (16|8)) & 0xff] variant.  */
  typedef unop_matcher<ZERO_EXTEND,
		       binop_matcher<AND,
			 table_access,
			 int_cst_matcher<minimal_term_info> > > shifted;
  /* zext (reg & cst) matcher.  Used for TABLE[val & 0xff] variant.  */
  typedef unop_matcher<ZERO_EXTEND,
		       binop_matcher<AND,
			 reg_matcher,
			 int_cst_matcher<minimal_term_info> > > noshift;

  std::map<rtx, typename TABLE_T::table_entry> table_alias;

  bool finalize (aes_table_ref<TABLE_T> &tref,
		 minimal_term_info &input_info,
		 minimal_term_info *shift_info = NULL,
		 minimal_term_info *mask_info = NULL)
    {
      gcc_assert (REG_P (input_info.src));
      gcc_assert (!shift_info || CONST_INT_P (shift_info->src));
      gcc_assert (!mask_info || CONST_INT_P (mask_info->src));

      unsigned HOST_WIDE_INT shift
	= shift_info ? UINTVAL (shift_info->src) : 0;
      unsigned HOST_WIDE_INT mask
	= mask_info ? UINTVAL (mask_info->src) : 0xFF;
      if (!verify_table_access (shift, mask, mask_info))
	return false;

      tref.insn = input_info.loc;
      tref.reg = input_info.src;
      tref.lsr_cst = shift;
      return true;
    }

  bool match (rtx_insn *insn, rtx index, aes_table_ref<TABLE_T> &tref)
    {
      direct::holder_type direct_res;
      if (direct::match (index, insn, direct_res))
	return finalize (tref, direct_res[0], &direct_res[1]);

      shifted::holder_type shifted_res;
      if (shifted::match (index, insn, shifted_res))
	return finalize (tref, shifted_res[0],
			 &shifted_res[1], &shifted_res[2]);

      noshift::holder_type noshift_res;
      return noshift::match (index, insn, noshift_res)
	&& finalize (tref, noshift_res[0], NULL, &noshift_res[1]);
    }

public:
  bool match (rtx_insn *insn, aes_table_ref<TABLE_T> &tref)
    {
      rtx mem = get_single_set_op<MEM> (insn);
      if (!mem && (mem = get_single_set_op<ZERO_EXTEND> (insn)))
	mem = XEXP (mem, 0);

      rtx dst = get_single_set_dst<REG> (insn);
      if (!mem || !MEM_P (mem) || !dst || GET_MODE (dst) != SImode)
	return false;

      simplified_mem_info info;
      if (!decompose_tref_mem_address (info, mem)
	  || !match (insn, info.index, tref))
	return false;

      typename TABLE_T::table_entry itable;
      if (!table_alias.count (info.base_reg))
	{
	  itable = check_table<TABLE_T> (mem);
	  if (itable == TABLE_T::BAD_TABLE)
	    return false;
	  table_alias[info.base_reg] = itable;
	}
      else
	itable = table_alias.at (info.base_reg);

      if (rtx_insn *out = get_possible_final_output<TABLE_T> (insn, dst,
							      tref.lsr_cst,
							      itable))
	{
	  tref.is_final = true;
	  tref.output_insn = out;
	  tref.output = NULL_RTX;
	}
      else
	{
	  tref.is_final = false;
	  tref.output_insn = insn;
	  tref.output = dst;
	}

      tref.itable = itable;
      return true;
    }
};

/* AES stage description.  Required for some specializations
   for curtain rounds.  */
typedef enum { aesINPUT, aesMIDDLE, aesFINAL } aes_stage;

/* AES entity description.  It can be both round or state inside round.
   It provides interface for unified analysis between blocks of 4 parts:
   round -> 4 states -> 4 * 4 arguments.  */
template<typename ENTRY_T, aes_stage STAGE, typename K>
struct aes_entity
{
  aes_key key;
  std::set<ENTRY_T> entries;
  rtx_insn *loc;

  aes_entity ()
    : key (NULL), loc (NULL)
    {}

  /* Push new entry to the entity.  */
  bool push_entry (const ENTRY_T &v)
    {
      if (entries.size () == 4)
	return false;

      entries.insert (v);
      return true;
    }

  /* The entities are sorted by key offset.  */
  bool operator < (const aes_entity &rhs) const
    {
      return key.offset < rhs.key.offset;
    }

  /* Verify that all of the entries are correct within their positions inside
     the entity.  */
  bool finalize ()
    {
      if (entries.size () != 4)
	return false;

      unsigned i = 0;
      const ENTRY_T *prev = NULL;
      for (typename std::set<ENTRY_T>::iterator it = entries.begin ();
	   it != entries.end (); prev = &*it++, ++i)
	if (!it->verify (prev, i))
	  return false;

      loc = entries.begin ()->loc;
      return true;
    }
};

/* Check the correctness of input regs permutations.  */
template<typename K>
static bool
check_input_regs (const std::vector<rtx> &curr,
		  const std::vector<rtx> &prev);

/* Specialize the function for AES encryption.  */
template<>
bool
check_input_regs<aes_encrypt_table> (const std::vector<rtx> &curr,
				 const std::vector<rtx> &prev)
{
  gcc_assert (curr.size () == 4 && prev.size () == 4);
  unsigned idx[4] = { 1, 2, 3, 0 };
  for (int i = 0; i < 4; ++i)
    if (curr[i] != prev[idx[i]])
      return false;
  return true;
}

/* Specialize the function for AES decryption.  */
template<>
bool
check_input_regs<aes_decrypt_table> (const std::vector<rtx> &curr,
				 const std::vector<rtx> &prev)
{
  gcc_assert (curr.size () == 4 && prev.size () == 4);
  unsigned idx[4] = { 3, 0, 1, 2 };
  for (int i = 0; i < 4; ++i)
    if (curr[i] != prev[idx[i]])
      return false;
  return true;
}

/* Basic descryption of state input.  */
template<aes_stage STAGE>
struct state_input
{
  typedef std::vector<rtx> type;

  static void finalize (type &in, rtx v)
    {
      in.push_back (v);
    }

  template<typename K>
  static bool verify (const type &lhs, const type &rhs)
    {
      return check_input_regs<K> (lhs, rhs);
    }
};

/* Input round state uses special input.  */
template<>
struct state_input<aesINPUT>
{
  typedef std::pair<rtx, unsigned HOST_WIDE_INT> type;

  static void finalize (type &in, const type &v)
    {
      in = v;
      // Order is inverted
      in.second -= 3;
    }

  template<typename>
  static bool verify (const type &lhs, const type &rhs)
    {
      return lhs.first == rhs.first
	&& lhs.second == rhs.second + 4;
    }
};

/* Basic descryption of state output.  */
template<aes_stage STAGE>
struct state_output
{
  typedef rtx type;

  static bool verify (const type &, const type &)
    {
      return true;
    }
};

/* Final round state generates special output.  */
template<>
struct state_output<aesFINAL>
{
  typedef std::pair<rtx, unsigned HOST_WIDE_INT> type;

  static bool verify (const type &lhs, const type &rhs)
    {
      return lhs.first == rhs.first
	&& lhs.second == rhs.second + 4;
    }
};

/* Basic descryption of round input.  */
template<aes_stage STAGE>
struct round_input
{
  typedef std::vector<rtx> type;
};

/* Input round uses special input just as its state.  */
template<>
struct round_input<aesINPUT>
{
  typedef std::pair<rtx, unsigned HOST_WIDE_INT> type;
};

/* Basic descryption of round output.  */
template<aes_stage STAGE>
struct round_output
{
  typedef std::vector<rtx> type;

  template<typename T>
  static void finalize (type &out, const T &v)
    {
      gcc_assert (v.size () == 4);
      for (typename T::const_iterator it = v.begin (); it != v.end (); ++it)
	out.push_back (it->output);
    }

  template<typename K>
  static void reorder (type &)
    {}
};

/* Reorder output for AES decryption: the order is changed compared to
   AES encryption.  */
template<>
template<>
void round_output<aesINPUT>::reorder<aes_decrypt_table> (type &out)
{
  gcc_assert (out.size () == 4);
  std::swap (out[1], out[3]);
}

template<>
template<>
void round_output<aesMIDDLE>::reorder<aes_decrypt_table> (type &out)
{
  round_output<aesINPUT>::reorder<aes_decrypt_table> (out);
}

/* Final round generates special output.  */
template<>
struct round_output<aesFINAL> : state_output<aesFINAL>
{
  template<typename T>
  static void finalize (type &out, const T &v)
    {
      gcc_assert (v.size () == 4);
      out = v.begin ()->output;
    }

  template<typename K>
  static void reorder (type &)
    {}
};

/* AES state descryption.  */
template<typename ENTRY_T, aes_stage STAGE, typename K>
struct aes_state : aes_entity<ENTRY_T, STAGE, K>
{
  typedef aes_entity<ENTRY_T, STAGE, K> base_entity;

  typename state_input<STAGE>::type input;
  typename state_output<STAGE>::type output;

  aes_state ()
    : base_entity ()
    {}

  void set_output (const typename state_output<STAGE>::type &o)
    {
      output = o;
    }

  bool push_entry (const ENTRY_T &v)
    {
      if (!v.is_key)
	return base_entity::push_entry (v);

      if (this->key.src)
	return false;

      this->key = v;
      return true;
    }

  /* Verify if the state is correct within its position in round.  */
  bool verify (const aes_state *prev, unsigned) const
    {
      if (!prev)
	return true;

      if (!this->key.has_same_origin (prev->key, this->loc)
	  || this->key.offset != prev->key.offset + 4
	  || BLOCK_FOR_INSN (this->loc) != BLOCK_FOR_INSN (prev->loc))
	return false;

      return state_input<STAGE>::template verify<K> (input, prev->input)
	&& state_output<STAGE>::verify (output, prev->output);
    }

  /* Check if the entries of the state are correct and finalize stored info.  */
  bool finalize ()
    {
      if (!base_entity::finalize ())
	return false;

      for (typename std::set<ENTRY_T>::iterator it = this->entries.begin ();
	   it != this->entries.end (); ++it)
	state_input<STAGE>::finalize (input, it->input ());

      return true;
    }
};

/* AES round descryption.  */
template<typename ENTRY_T, aes_stage STAGE, typename K>
struct aes_round : aes_entity<aes_state<ENTRY_T, STAGE, K>, STAGE, K>
{
  typedef aes_entity<aes_state<ENTRY_T, STAGE, K>, STAGE, K> base_entity;

  typename round_input<STAGE>::type input;
  typename round_output<STAGE>::type output;

  /* Check if the states are correct and finalize stored info.  */
  bool finalize ()
    {
      if (!base_entity::finalize ())
	return false;

      input = this->entries.begin ()->input;
      this->key = this->entries.begin ()->key;

      round_output<STAGE>::finalize (output, this->entries);
      round_output<STAGE>::template reorder<K> (output);

      return true;
    }
};

template<typename T>
class aes_optimizer;

/* AES round input info.  Used to find and store info about
   table references.

   Must be inited and finalized before and after usage.  */
template<typename T>
struct round_input_info : state_input_info
{
  typedef typename aes_optimizer<T>::table_ref_map tref_map;

  round_input_info ()
    {}
  round_input_info (rtx_insn *insn, const aes_table_ref<T> *tref)
    : state_input_info (mem_term_info (insn, NULL_RTX)), tref (tref)
    {}
  round_input_info (const aes_key &k)
    : state_input_info (k)
    {}

  rtx input () const
    {
      return tref->reg;
    }

  rtx output () const
    {
      return tref->output;
    }

  /* Table references are sorted by shift constants.
     TODO: probably sort by key offset?  */
  bool operator < (const round_input_info &rhs) const
    {
      return tref->lsr_cst > rhs.tref->lsr_cst;
    }

  bool verify (const round_input_info *prev, unsigned i) const
    {
      return state_input_info::verify (prev) && tref->verify (i);
    }

  static bool finalize (rtx_insn *insn, round_input_info *m)
    {
      if (checked_p->count (insn))
	return false;

      typename tref_map::const_iterator it = table_refs_p->find (insn);
      if (it == table_refs_p->end ())
	return false;

      m[0] = round_input_info (insn, &it->second);
      return true;
    }

  const aes_table_ref<T> *tref;

  static const tref_map *table_refs_p;
  static const std::set<rtx_insn *> *checked_p;

  /* Store lookup table references.  */
  static void init (const tref_map &t, const std::set<rtx_insn *> &c)
    {
      gcc_assert (!table_refs_p && !checked_p);
      table_refs_p = &t;
      checked_p = &c;
    }

  /* Remove lookup table references.  */
  static void fin ()
    {
      gcc_assert (table_refs_p && checked_p);
      table_refs_p = NULL;
      checked_p = NULL;
    }
};

template<typename T>
const typename aes_optimizer<T>::table_ref_map *
round_input_info<T>::table_refs_p = NULL;

template<typename T>
const std::set<rtx_insn *> *
round_input_info<T>::checked_p = NULL;

/* AES encryption/decryption optimizer.  */
template<typename T>
class aes_optimizer
{
public:
  typedef std::map<rtx_insn *, aes_table_ref<T> > table_ref_map;

  /* AES states typedefs.  */
  typedef aes_state<input_info, aesINPUT, T> aes_input_state;
  typedef aes_state<round_input_info<T>, aesMIDDLE, T> aes_body_state;
  typedef aes_state<round_input_info<T>, aesFINAL, T> aes_final_state;

  /* AES rounds typedefs.  */
  typedef aes_round<input_info, aesINPUT, T> aes_input_round;
  typedef aes_round<round_input_info<T>, aesMIDDLE, T> aes_body_round;
  typedef aes_round<round_input_info<T>, aesFINAL, T> aes_final_round;

  bool run ();

private:
  bool collect_aes_lookup_tables ();
  bool form_rounds ();
  bool find_aes_init_round ();
  bool collect_state (rtx_insn * insn, aes_body_state &state,
		      std::set<rtx_insn *> &checked);
  bool find_aes_rounds ();
  bool collect_final_round (rtx_insn *insn, aes_final_state &state,
			    std::set<rtx_insn *> &checked);
  bool find_aes_final_round ();
  bool check_aes_pattern ();
  void erase_unused_rounds (std::set<const std::vector<rtx> *> &used);

  bool gen_aes_code ();
  bool gen_init_round ();
  bool gen_round (const aes_body_round &round);
  bool gen_final_round ();

  rtx gen_or_get_vreg (const std::vector<rtx> &vec);
  rtx get_vreg (const std::vector<rtx> &vec);
  rtx gen_vreg (const std::vector<rtx> &vec);

  table_ref_map table_refs;
  table_ref_map final_table_refs;

  aes_input_round input_round;
  std::map<std::vector<rtx>, aes_body_round> rounds;
  aes_final_round final_round;

  std::map<std::vector<rtx>, rtx> vec_regs;
  std::vector<rtx_insn*> to_delete;
};

/* Find all the AES table references in function.  */
template<typename T>
bool
aes_optimizer<T>::collect_aes_lookup_tables ()
{
  basic_block bb;
  rtx_insn *insn;

  tref_matcher<T> m;
  FOR_EACH_BB_FN (bb, cfun)
    FOR_BB_INSNS (bb, insn)
      {
	aes_table_ref<T> tref;
	if (!m.match (insn, tref))
	  continue;

	if (!tref.is_final)
	  table_refs[insn] = tref;
	else
	  final_table_refs[tref.output_insn] = tref;
      }

  return !table_refs.empty () && !final_table_refs.empty ();
}

/* Helper function to match all the permutations of five arg
   calculations.  */
template<rtx_code OP1, typename TERM, rtx_code OP2 = OP1,
	 bool store_ops = false>
struct five_args_calc_matcher
{
  /* Helper for matching (op1 * op2).  */
  typedef binop_matcher<OP1, TERM, TERM, false, store_ops, OP2> two_args_block;
  /* Helper for matching (op1 * (op2 * op3)).  */
  typedef binop_matcher<OP1, two_args_block, TERM,
			true, store_ops, OP2> three_args_block;
  /* Helper for matching ((op1 * op2) * (op3 * op4)).  */
  typedef binop_matcher<OP1, two_args_block, two_args_block,
			false, store_ops, OP2> opt_four_args_block;
  /* Helper for matching (op1 * (op2 * (op3 * op4))).  */
  typedef binop_matcher<OP1, three_args_block, TERM,
			       true, store_ops, OP2> linear_four_args_block;

  /* Match the (op1 * ((op2 * op3) * (op4 * op5))) variant.  */
  typedef binop_matcher<OP1, opt_four_args_block, TERM,
			true, store_ops, OP2> opt_op_term;
  /* Match the ((op1 * op2) * (op3 * (op4 * op5))) variant.  */
  typedef binop_matcher<OP1, three_args_block, two_args_block,
			true, store_ops, OP2> three_op_two;
  /* Match the (op1 * (op2 * (op3 * (op4 * op5)))) variant.  */
  typedef binop_matcher<OP1, linear_four_args_block, TERM,
			true, store_ops, OP2> fully_linear;

  static const int holder_size = fully_linear::holder_size;
  static const int op_num = fully_linear::op_num;
  typedef typename fully_linear::term_type term_type;
  typedef typename fully_linear::holder_type holder_type;

  static rtx_insn* match (rtx_insn *insn, holder_type &m, unsigned depth = 1)
    {
      for (rtx dst = get_single_set_dst (insn); depth && insn && dst;
	   insn = get_single_use_insn (insn, dst),
	   dst = insn ? get_single_set_dst (insn) : NULL_RTX,
	   --depth)
	if (opt_op_term::match (insn, m) || three_op_two::match (insn, m)
	    || fully_linear::match (insn, m))
	  return insn;
      return NULL;
    }
};

/* Match the AES key.  */
struct key_matcher : matcher_term<aes_key>
{
  static bool match (rtx_insn *insn, holder_type &m)
    {
      mem_matcher::holder_type info;
      if (!mem_matcher::match (insn, info))
	return false;

      m[0] = info[0];
      return true;
    }
};

/* Matcher term for state input.  */
template<typename T>
struct state_input_term : matcher_term<T>
{
  typedef typename matcher_term<T>::holder_type holder_type;

  static bool match (rtx, rtx_insn *, holder_type &)
    {
      return false;
    }

  static bool match (rtx_insn *insn, holder_type &m)
    {
      key_matcher::holder_type k;
      if (key_matcher::match (insn, k))
	{
	  m[0] = k[0];
	  return true;
	}

      return matcher_term<T>::term_type::finalize (insn, m);
    }
};

/* Fill state from args.  */
template <typename STATE, typename T>
static bool
finalize_input (const T (&args)[5], STATE &state)
{
  for (unsigned i = 0; i < 5; ++i)
    if (!state.push_entry (args[i]))
      return false;

  return state.finalize ();
}

/* Construct input state.  */
template<typename T>
static bool
form_input (rtx_insn *insn, T &state)
{
  typedef five_args_calc_matcher<XOR, state_input_term<input_info> >
    matcher;

  matcher::holder_type m;
  if (!matcher::match (insn, m) || !finalize_input (m, state))
    return false;

  /* TODO: probably should not be set here.  */
  state.set_output (SET_DEST (single_set (insn)));
  return true;
}

/* Get definitions chain for the reg being used in the insn.  */
static df_link *
get_defs (rtx_insn *insn, rtx reg)
{
  df_link *ref_chain = get_def_chain (insn, reg);
  gcc_assert (ref_chain);

  for (df_link *ref_link = ref_chain; ref_link; ref_link = ref_link->next)
    if (!check_def_chain_ref (ref_link->ref, reg))
      return NULL;

  return ref_chain;
}

/* Find AES init round.  To do this, find the table references that depends on
   two definitions.  One of them is our input.  */
template<typename T>
bool
aes_optimizer<T>::find_aes_init_round ()
{
  std::set<rtx_insn *> checked;

  for (typename table_ref_map::iterator it = table_refs.begin (),
       end = table_refs.end (); it != end; ++it)
    for (df_link *def = get_defs (it->second.insn, it->second.reg);
	 def; def = def->next)
      {
	rtx_insn *def_insn = DF_REF_INSN (def->ref);
	if (checked.count (def_insn))
	  continue;

	aes_input_state input_state;
	if (form_input (def_insn, input_state)
	    && !input_round.push_entry (input_state))
	  return false;

	checked.insert (def_insn);
      }

  return input_round.finalize ();
}

/* Collect AES inner state.  */
template<typename T>
bool
aes_optimizer<T>::collect_state (rtx_insn *insn, aes_body_state &state,
				 std::set<rtx_insn*> &checked)
{
  typedef round_input_info<T> term_info;
  typedef five_args_calc_matcher<XOR, state_input_term<term_info> > matcher;

  typename matcher::holder_type m;
  term_info::init (table_refs, checked);
  rtx_insn *match_entry = matcher::match (insn, m, 3);
  term_info::fin ();

  if (!match_entry || !finalize_input (m, state))
    return false;

  /* TODO: probably should not be set here.  */
  state.set_output (SET_DEST (single_set (match_entry)));
  for (unsigned i = 0; i < 5; ++i)
    if (!m[i].is_key)
      checked.insert (m[i].tref->output_insn);

  return true;
}

/* Simple sorter to link rounds by their registers.  */
struct reg_comp
{
  bool operator () (rtx lhs, rtx rhs) const
    {
      return REGNO (lhs) < REGNO (rhs);
    }
};

/* Find AES inner rounds.  */
template<typename T>
bool
aes_optimizer<T>::find_aes_rounds ()
{
  typedef std::set<rtx, reg_comp> input_key;

  std::set<rtx_insn*> checked;
  std::map<input_key, aes_body_round> candidate_rounds;
  for (typename table_ref_map::iterator it = table_refs.begin (),
       end = table_refs.end (); it != end; ++it)
    {
      rtx_insn *insn = it->first;
      if (checked.count (insn))
	continue;

      rtx_insn *use = get_single_use_insn (insn, SET_DEST (single_set (insn)));
      if (!use)
	continue;

      aes_body_state state;
      if (!collect_state (use, state, checked))
	continue;

      /* Sort the input so we can found the corresponding state.  */
      input_key input (state.input.begin (), state.input.end ());
      candidate_rounds[input].push_entry (state);
    }

  for (typename std::map<input_key, aes_body_round>::iterator
       it = candidate_rounds.begin ();
       it != candidate_rounds.end (); ++it)
    if (it->second.finalize ())
      rounds[it->second.input] = it->second;

  return !rounds.empty ();
}

template<typename T>
struct final_state_matcher;

/* AES encrypt matcher requires additional check on key calculations
   due to possible optimizations.  */
template<>
struct final_state_matcher<aes_encrypt_table>
{
  typedef round_input_info<aes_encrypt_table> term_info;
  typedef five_args_calc_matcher<XOR, state_input_term<term_info>, IOR, true>
    matcher;
  typedef typename matcher::term_type
    holder_type[matcher::holder_size - matcher::op_num];

  static rtx_insn *match (rtx_insn *insn, holder_type &m, unsigned depth)
    {
      matcher::holder_type inner_m;
      rtx_insn *res = matcher::match (insn, inner_m, depth);
      if (!res)
	return NULL;

      /* Run pre-order traversal of the operands to check the correctness
	 of key usage.  */
      gcc_assert (inner_m[0].is_op);
      unsigned pos = 0;
      if (!check_key_calculations (inner_m, pos))
	return NULL;
      gcc_assert (pos == (matcher::holder_size - 1));

      unsigned idx = 0;
      for (unsigned i = 0; i < matcher::holder_size; ++i)
	if (!inner_m[i].is_op)
	  m[idx++] = inner_m[i];

      gcc_assert (idx == 5);
      return res;
    }

  static bool check_key_calculations (const matcher::holder_type &m,
				      unsigned &idx,
				      bool failure_on_key = false)
    {
      gcc_assert (idx < matcher::holder_size);
      if (!m[idx].is_op)
	return !(failure_on_key && m[idx].is_key);

      failure_on_key |= (GET_CODE (m[idx].src) == IOR);
      return check_key_calculations (m, ++idx, failure_on_key)
	&& check_key_calculations (m, ++idx, failure_on_key);
    }
};


/* The final state is simple wrapper since no additional checks are required
   here.  */
template<>
struct final_state_matcher<aes_decrypt_table>
{
  typedef round_input_info<aes_decrypt_table> term_info;
  typedef five_args_calc_matcher<XOR, state_input_term<term_info> > matcher;
  typedef typename matcher::holder_type holder_type;

  static rtx_insn *match (rtx_insn *insn, holder_type &m, unsigned depth)
    {
      return matcher::match (insn, m, depth);
    }
};

/* Match the AES final state.  */
template<typename T>
bool
aes_optimizer<T>::collect_final_round (rtx_insn *insn, aes_final_state &state,
				       std::set<rtx_insn*> &checked)
{
  typedef final_state_matcher<T> matcher_wrapper;

  typename matcher_wrapper::holder_type m;
  matcher_wrapper::term_info::init (final_table_refs, checked);
  rtx_insn *match_entry = matcher_wrapper::match (insn, m, 3);
  matcher_wrapper::term_info::fin ();

  rtx dst;
  if (!match_entry || !(dst = get_single_set_dst (match_entry))
      || !finalize_input (m, state))
    return false;

  rtx src;
  if (!(match_entry = get_single_use_insn (match_entry, dst))
      || !(check_simple_op<BSWAP> (match_entry, src, dst))
      || !dst)
    return false;

  std::pair<rtx, unsigned HOST_WIDE_INT> output;
  if (!(match_entry = get_single_use_insn (match_entry, dst))
      || !(dst = get_single_set_dst<MEM> (match_entry))
      || !decompose_mem (dst, output.first, output.second))
    return false;

  to_delete.push_back (match_entry);
  state.set_output (output);
  for (unsigned i = 0; i < 5; ++i)
    if (!m[i].is_key)
      checked.insert (m[i].tref->output_insn);

  return true;
}

/* Find the final round.  */
template<typename T>
bool
aes_optimizer<T>::find_aes_final_round ()
{
  std::set<rtx_insn*> checked;
  for (typename table_ref_map::iterator it = final_table_refs.begin (),
       end = final_table_refs.end (); it != end; ++it)
    {
      rtx_insn *insn = it->first;

      if (checked.count (insn))
	continue;

      rtx_insn *use = get_single_use_insn (insn, SET_DEST (single_set (insn)));
      if (!use)
	continue;

      aes_final_state state;
      if (collect_final_round (use, state, checked))
	final_round.push_entry (state);
    }

  return final_round.finalize ();
}

template<typename T>
bool
aes_optimizer<T>::form_rounds ()
{
  return find_aes_final_round ()
    && find_aes_init_round ()
    && find_aes_rounds ();
}

template<typename T>
void
aes_optimizer<T>::erase_unused_rounds (std::set<const std::vector<rtx> *> &used)
{
  if (used.size () == rounds.size ())
    return;

  for (typename std::map<std::vector<rtx>, aes_body_round>::iterator
       it = rounds.begin (), next = it,
       end = rounds.end (); it != end; it = next)
    {
      ++next;
      if (!used.count (&it->first))
	rounds.erase (it);
    }
}

/* Find round starts and link them together.  */
template<typename T>
bool
aes_optimizer<T>::check_aes_pattern ()
{
  std::set<const std::vector<rtx> *> checked;

  typename std::map<std::vector<rtx>, aes_body_round>::iterator fit
    = rounds.find (input_round.output);

  bool to_final = false;
  while (fit != rounds.end () && !checked.count (&fit->first))
    {
      checked.insert (&fit->first);

      if (fit->second.output == final_round.input)
	to_final = true;

      fit = rounds.find (fit->second.output);
    }

  if (!to_final)
    return false;

  erase_unused_rounds (checked);

  return true;
}

static bool
gen_insns (const rtx patterns[4], rtx_insn *loc)
{
  start_sequence ();
  for (unsigned i = 0; i < 4; ++i)
    {
      rtx_insn *insn = emit_insn (patterns[i]);
      if (recog_memoized (insn) < 0)
	{
	  end_sequence ();
	  return false;
	}
    }

  rtx_insn *seq = get_insns ();
  end_sequence ();
  emit_insn_after (seq, loc);

  return true;
}

static rtx
gen_offset_access (rtx base, unsigned HOST_WIDE_INT offset)
{
  if (!offset)
    return base;

  machine_mode mode = GET_MODE (base);
  return gen_rtx_PLUS (mode, base, gen_rtx_CONST_INT (mode, offset));
}

template<typename T>
rtx
aes_optimizer<T>::get_vreg (const std::vector<rtx> &vec)
{
  std::map<std::vector<rtx>, rtx>::iterator fit = vec_regs.find (vec);
  if (fit != vec_regs.end ())
    return fit->second;

  return 0;
}

template<typename T>
rtx
aes_optimizer<T>::gen_vreg (const std::vector<rtx> &vec)
{
  machine_mode vmode = targetm.get_v16qi_mode ();
  rtx vreg = gen_reg_rtx (vmode);
  vec_regs.insert (std::make_pair (vec, vreg));

  return vreg;
}

template<typename T>
rtx
aes_optimizer<T>::gen_or_get_vreg (const std::vector<rtx> &vec)
{
  rtx vreg = get_vreg (vec);
  if (!vreg)
    vreg = gen_vreg (vec);

  return vreg;
}

template<typename T>
static rtx
gen_aes_single_round (rtx vout, rtx vreg, rtx vkey);
template<typename T>
static rtx
gen_aes_mix_columns (rtx vreg, rtx vin);

template<>
rtx
gen_aes_single_round<aes_encrypt_table> (rtx vout, rtx vreg, rtx vkey)
{
  return targetm.gen_aesev16qi (vout, vreg, vkey);
}

template<>
rtx
gen_aes_mix_columns<aes_encrypt_table> (rtx vreg, rtx vin)
{
  return targetm.gen_aesmcv16qi (vreg, vin);
}

template<>
rtx
gen_aes_single_round<aes_decrypt_table> (rtx vout, rtx vreg, rtx vkey)
{
  return targetm.gen_aesdv16qi (vout, vreg, vkey);
}

template<>
rtx
gen_aes_mix_columns<aes_decrypt_table> (rtx vreg, rtx vin)
{
  return targetm.gen_aesimcv16qi (vreg, vin);
}

template<typename T>
bool
aes_optimizer<T>::gen_init_round ()
{
  rtx_insn *loc = input_round.loc;

  machine_mode vmode = targetm.get_v16qi_mode ();

  rtx vreg = gen_reg_rtx (vmode);
  rtx vkey = gen_reg_rtx (vmode);
  rtx vout = gen_vreg (input_round.output);

  rtx buf = input_round.input.first;
  rtx key = gen_offset_access (input_round.key.src, input_round.key.offset);

  rtx vload_pat = gen_rtx_SET (vreg,
			       gen_rtx_MEM (vmode, buf));
  rtx vkey_load_pat = gen_rtx_SET (vkey,
				   gen_rtx_MEM (vmode, key));
  rtx vrev_pat = targetm.gen_rev32v16qi (vkey, vkey);
  rtx vaes_pat = gen_aes_single_round<T> (vout, vreg, vkey);

  const rtx patterns[4] = {vload_pat, vkey_load_pat, vrev_pat, vaes_pat};

  return gen_insns (patterns, loc);
}

template<typename T>
bool
aes_optimizer<T>::gen_round (const aes_body_round &round)
{
  rtx_insn *loc = round.loc;

  machine_mode vmode = targetm.get_v16qi_mode ();

  rtx vreg = gen_reg_rtx (vmode);
  rtx vkey = gen_reg_rtx (vmode);
  rtx vin  = gen_or_get_vreg (round.input);
  rtx vout = gen_or_get_vreg (round.output);

  rtx key = gen_offset_access (round.key.src, round.key.offset);

  rtx vkey_load_pat = gen_rtx_SET (vkey,
				   gen_rtx_MEM (vmode, key));
  rtx vrev_pat = targetm.gen_rev32v16qi (vkey, vkey);
  rtx vmix_pat = gen_aes_mix_columns<T> (vreg, vin);
  rtx vaes_pat = gen_aes_single_round<T> (vout, vreg, vkey);

  const rtx patterns[4] = {vkey_load_pat, vrev_pat, vmix_pat, vaes_pat};

  return gen_insns (patterns, loc);
}

template<typename T>
bool
aes_optimizer<T>::gen_final_round ()
{
  rtx_insn *loc = final_round.loc;

  machine_mode vmode = targetm.get_v16qi_mode ();

  rtx vreg = gen_reg_rtx (vmode);
  rtx vkey = gen_reg_rtx (vmode);
  rtx vin = get_vreg (final_round.input);

  gcc_assert (vin);

  rtx buf = final_round.output.first;
  rtx key = gen_offset_access (final_round.key.src, final_round.key.offset);

  rtx vkey_load_pat = gen_rtx_SET (vkey,
				   gen_rtx_MEM (vmode, key));
  rtx vrev_pat = targetm.gen_rev32v16qi (vkey, vkey);
  rtx vxor_pat = gen_rtx_SET (vreg, gen_rtx_XOR (vmode, vin, vkey));
  rtx vstore_pat = gen_rtx_SET (gen_rtx_MEM (vmode, buf), vreg);

  const rtx patterns[4] = {vkey_load_pat, vrev_pat, vxor_pat, vstore_pat};

  return gen_insns (patterns, loc);
}

template<typename T>
bool
aes_optimizer<T>::gen_aes_code ()
{
  if (!gen_init_round ())
    return false;

  for (typename std::map<std::vector<rtx>, aes_body_round>::iterator
       it = rounds.begin (), end = rounds.end (); it != end; ++it)
    {
      if (!gen_round (it->second))
	return false;
    }

  if (!gen_final_round ())
    return false;

  for (std::vector<rtx_insn*>::iterator it = to_delete.begin (),
       end = to_delete.end (); it != end; ++it)
    SET_INSN_DELETED (*it);

  return true;
}

template<typename T>
bool
aes_optimizer<T>::run ()
{
  return collect_aes_lookup_tables ()
    && form_rounds ()
    && check_aes_pattern ()
    && gen_aes_code ();
}

static unsigned int
crypto_acceleration ()
{
  aes_optimizer<aes_encrypt_table> enc;
  aes_optimizer<aes_decrypt_table> dec;
  enc.run ();
  dec.run ();

  return 0;
}

static void
init_df ()
{
  df_set_flags (DF_RD_PRUNE_DEAD_DEFS);
  df_chain_add_problem (DF_UD_CHAIN + DF_DU_CHAIN);
  df_mir_add_problem ();
  df_live_add_problem ();
  df_live_set_all_dirty ();
  df_analyze ();
  df_set_flags (DF_DEFER_INSN_RESCAN);
}

namespace {

const pass_data pass_data_crypto_accel =
{
  RTL_PASS,	   // type
  "crypto_accel",  // name
  OPTGROUP_NONE,   // optinfo_flags
  TV_CRYPTO_ACCEL, // tv_id
  PROP_cfglayout,  // properties_required
  0,		   // properties_provided
  0,		   // properties_destroyed
  0,		   // todo_flags_start
  TODO_df_finish,  // todo_flags_finish
};

class pass_crypto_accel : public rtl_opt_pass
{
public:
  pass_crypto_accel (gcc::context *ctxt)
    : rtl_opt_pass (pass_data_crypto_accel, ctxt)
  {}

  /* opt_pass methods: */
  virtual bool gate (function *)
    {
      if (flag_crypto_accel_aes <= 0 || optimize < 1)
	return false;
      return targetm.get_v16qi_mode
	&& targetm.gen_rev32v16qi
	&& targetm.gen_aesev16qi
	&& targetm.gen_aesmcv16qi;
    }

  virtual unsigned int execute (function *)
    {
      init_df ();
      return crypto_acceleration ();
    }
}; // class pass_crypto_accel

} // anon namespace

rtl_opt_pass *
make_pass_crypto_accel (gcc::context *ctxt)
{
  return new pass_crypto_accel (ctxt);
}
